import{s as me,o as he,n as ce}from"../chunks/scheduler.182ea377.js";import{S as _e,i as ge,g as d,s as o,r as v,A as $e,h as u,f as t,c as r,j as A,u as y,x as k,k as S,y as m,a,v as T,d as D,t as M,w as P}from"../chunks/index.1ea56653.js";import{H as ae,g as we}from"../chunks/Heading.71ac6e9f.js";import{T as be}from"../chunks/Tip.67abc998.js";import{D as oe}from"../chunks/Docstring.59aaf0cd.js";import{C as ve}from"../chunks/CodeBlock.82c2494b.js";import{E as ye}from"../chunks/ExampleCodeBlock.95ae1d71.js";function Te(V){let s,g='Make sure to check out the Schedulers <a href="../../using-diffusers/schedulers">guide</a> to learn how to explore the tradeoff between scheduler speed and quality, and see the <a href="../../using-diffusers/loading#reuse-components-across-pipelines">reuse components across pipelines</a> section to learn how to efficiently load the same components into multiple pipelines.';return{c(){s=d("p"),s.innerHTML=g},l(l){s=u(l,"P",{"data-svelte-h":!0}),k(s)!=="svelte-1wmc0l4"&&(s.innerHTML=g)},m(l,p){a(l,s,p)},p:ce,d(l){l&&t(s)}}}function De(V){let s,g="Example:",l,p,c;return p=new ve({props:{code:"ZnJvbSUyMGRpZmZ1c2VycyUyMGltcG9ydCUyMERpZmZ1c2lvblBpcGVsaW5lJTBBZnJvbSUyMHNjaXB5LmlvLndhdmZpbGUlMjBpbXBvcnQlMjB3cml0ZSUwQSUwQW1vZGVsX2lkJTIwJTNEJTIwJTIyaGFybW9uYWklMkZtYWVzdHJvLTE1MGslMjIlMEFwaXBlJTIwJTNEJTIwRGlmZnVzaW9uUGlwZWxpbmUuZnJvbV9wcmV0cmFpbmVkKG1vZGVsX2lkKSUwQXBpcGUlMjAlM0QlMjBwaXBlLnRvKCUyMmN1ZGElMjIpJTBBJTBBYXVkaW9zJTIwJTNEJTIwcGlwZShhdWRpb19sZW5ndGhfaW5fcyUzRDQuMCkuYXVkaW9zJTBBJTBBJTIzJTIwVG8lMjBzYXZlJTIwbG9jYWxseSUwQWZvciUyMGklMkMlMjBhdWRpbyUyMGluJTIwZW51bWVyYXRlKGF1ZGlvcyklM0ElMEElMjAlMjAlMjAlMjB3cml0ZShmJTIybWFlc3Ryb190ZXN0XyU3QmklN0Qud2F2JTIyJTJDJTIwcGlwZS51bmV0LnNhbXBsZV9yYXRlJTJDJTIwYXVkaW8udHJhbnNwb3NlKCkpJTBBJTBBJTIzJTIwVG8lMjBkaXNsYXklMjBpbiUyMGdvb2dsZSUyMGNvbGFiJTBBaW1wb3J0JTIwSVB5dGhvbi5kaXNwbGF5JTIwYXMlMjBpcGQlMEElMEFmb3IlMjBhdWRpbyUyMGluJTIwYXVkaW9zJTNBJTBBJTIwJTIwJTIwJTIwZGlzcGxheShpcGQuQXVkaW8oYXVkaW8lMkMlMjByYXRlJTNEcGlwZS51bmV0LnNhbXBsZV9yYXRlKSk=",highlighted:`<span class="hljs-keyword">from</span> diffusers <span class="hljs-keyword">import</span> DiffusionPipeline
<span class="hljs-keyword">from</span> scipy.io.wavfile <span class="hljs-keyword">import</span> write

model_id = <span class="hljs-string">&quot;harmonai/maestro-150k&quot;</span>
pipe = DiffusionPipeline.from_pretrained(model_id)
pipe = pipe.to(<span class="hljs-string">&quot;cuda&quot;</span>)

audios = pipe(audio_length_in_s=<span class="hljs-number">4.0</span>).audios

<span class="hljs-comment"># To save locally</span>
<span class="hljs-keyword">for</span> i, audio <span class="hljs-keyword">in</span> <span class="hljs-built_in">enumerate</span>(audios):
    write(<span class="hljs-string">f&quot;maestro_test_<span class="hljs-subst">{i}</span>.wav&quot;</span>, pipe.unet.sample_rate, audio.transpose())

<span class="hljs-comment"># To dislay in google colab</span>
<span class="hljs-keyword">import</span> IPython.display <span class="hljs-keyword">as</span> ipd

<span class="hljs-keyword">for</span> audio <span class="hljs-keyword">in</span> audios:
    display(ipd.Audio(audio, rate=pipe.unet.sample_rate))`,wrap:!1}}),{c(){s=d("p"),s.textContent=g,l=o(),v(p.$$.fragment)},l(i){s=u(i,"P",{"data-svelte-h":!0}),k(s)!=="svelte-11lpom8"&&(s.textContent=g),l=r(i),y(p.$$.fragment,i)},m(i,h){a(i,s,h),a(i,l,h),T(p,i,h),c=!0},p:ce,i(i){c||(D(p.$$.fragment,i),c=!0)},o(i){M(p.$$.fragment,i),c=!1},d(i){i&&(t(s),t(l)),P(p,i)}}}function Me(V){let s,g,l,p,c,i,h,re='<a href="https://github.com/Harmonai-org/sample-generator" rel="nofollow">Dance Diffusion</a> is by Zach Evans.',z,B,le='Dance Diffusion is the first in a suite of generative audio tools for producers and musicians released by <a href="https://github.com/Harmonai-org" rel="nofollow">Harmonai</a>.',L,J,H,G,O,f,j,K,X,pe="Pipeline for audio generation.",ee,E,de=`This model inherits from <a href="/docs/diffusers/main/en/api/pipelines/overview#diffusers.DiffusionPipeline">DiffusionPipeline</a>. Check the superclass documentation for the generic methods
implemented for all pipelines (downloading, saving, running on a particular device, etc.).`,ne,$,I,te,N,ue="The call function to the pipeline for generation.",se,x,Y,Z,R,w,C,ie,U,fe="Output class for audio pipelines.",Q,W,q;return c=new ae({props:{title:"Dance Diffusion",local:"dance-diffusion",headingTag:"h1"}}),J=new be({props:{$$slots:{default:[Te]},$$scope:{ctx:V}}}),G=new ae({props:{title:"DanceDiffusionPipeline",local:"diffusers.DanceDiffusionPipeline",headingTag:"h2"}}),j=new oe({props:{name:"class diffusers.DanceDiffusionPipeline",anchor:"diffusers.DanceDiffusionPipeline",parameters:[{name:"unet",val:""},{name:"scheduler",val:""}],parametersDescription:[{anchor:"diffusers.DanceDiffusionPipeline.unet",description:`<strong>unet</strong> (<a href="/docs/diffusers/main/en/api/models/unet#diffusers.UNet1DModel">UNet1DModel</a>) &#x2014;
A <code>UNet1DModel</code> to denoise the encoded audio.`,name:"unet"},{anchor:"diffusers.DanceDiffusionPipeline.scheduler",description:`<strong>scheduler</strong> (<a href="/docs/diffusers/main/en/api/schedulers/overview#diffusers.SchedulerMixin">SchedulerMixin</a>) &#x2014;
A scheduler to be used in combination with <code>unet</code> to denoise the encoded audio latents. Can be one of
<a href="/docs/diffusers/main/en/api/schedulers/ipndm#diffusers.IPNDMScheduler">IPNDMScheduler</a>.`,name:"scheduler"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py#L28"}}),I=new oe({props:{name:"__call__",anchor:"diffusers.DanceDiffusionPipeline.__call__",parameters:[{name:"batch_size",val:": int = 1"},{name:"num_inference_steps",val:": int = 100"},{name:"generator",val:": typing.Union[torch._C.Generator, typing.List[torch._C.Generator], NoneType] = None"},{name:"audio_length_in_s",val:": typing.Optional[float] = None"},{name:"return_dict",val:": bool = True"}],parametersDescription:[{anchor:"diffusers.DanceDiffusionPipeline.__call__.batch_size",description:`<strong>batch_size</strong> (<code>int</code>, <em>optional</em>, defaults to 1) &#x2014;
The number of audio samples to generate.`,name:"batch_size"},{anchor:"diffusers.DanceDiffusionPipeline.__call__.num_inference_steps",description:`<strong>num_inference_steps</strong> (<code>int</code>, <em>optional</em>, defaults to 50) &#x2014;
The number of denoising steps. More denoising steps usually lead to a higher-quality audio sample at
the expense of slower inference.`,name:"num_inference_steps"},{anchor:"diffusers.DanceDiffusionPipeline.__call__.generator",description:`<strong>generator</strong> (<code>torch.Generator</code>, <em>optional</em>) &#x2014;
A <a href="https://pytorch.org/docs/stable/generated/torch.Generator.html" rel="nofollow"><code>torch.Generator</code></a> to make
generation deterministic.`,name:"generator"},{anchor:"diffusers.DanceDiffusionPipeline.__call__.audio_length_in_s",description:`<strong>audio_length_in_s</strong> (<code>float</code>, <em>optional</em>, defaults to <code>self.unet.config.sample_size/self.unet.config.sample_rate</code>) &#x2014;
The length of the generated audio sample in seconds.`,name:"audio_length_in_s"},{anchor:"diffusers.DanceDiffusionPipeline.__call__.return_dict",description:`<strong>return_dict</strong> (<code>bool</code>, <em>optional</em>, defaults to <code>True</code>) &#x2014;
Whether or not to return a <a href="/docs/diffusers/main/en/api/pipelines/dance_diffusion#diffusers.AudioPipelineOutput">AudioPipelineOutput</a> instead of a plain tuple.`,name:"return_dict"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/dance_diffusion/pipeline_dance_diffusion.py#L48",returnDescription:`<script context="module">export const metadata = 'undefined';<\/script>


<p>If <code>return_dict</code> is <code>True</code>, <a
  href="/docs/diffusers/main/en/api/pipelines/dance_diffusion#diffusers.AudioPipelineOutput"
>AudioPipelineOutput</a> is returned, otherwise a <code>tuple</code> is
returned where the first element is a list with the generated audio.</p>
`,returnType:`<script context="module">export const metadata = 'undefined';<\/script>


<p><a
  href="/docs/diffusers/main/en/api/pipelines/dance_diffusion#diffusers.AudioPipelineOutput"
>AudioPipelineOutput</a> or <code>tuple</code></p>
`}}),x=new ye({props:{anchor:"diffusers.DanceDiffusionPipeline.__call__.example",$$slots:{default:[De]},$$scope:{ctx:V}}}),Z=new ae({props:{title:"AudioPipelineOutput",local:"diffusers.AudioPipelineOutput",headingTag:"h2"}}),C=new oe({props:{name:"class diffusers.AudioPipelineOutput",anchor:"diffusers.AudioPipelineOutput",parameters:[{name:"audios",val:": ndarray"}],parametersDescription:[{anchor:"diffusers.AudioPipelineOutput.audios",description:`<strong>audios</strong> (<code>np.ndarray</code>) &#x2014;
List of denoised audio samples of a NumPy array of shape <code>(batch_size, num_channels, sample_rate)</code>.`,name:"audios"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/pipeline_utils.py#L125"}}),{c(){s=d("meta"),g=o(),l=d("p"),p=o(),v(c.$$.fragment),i=o(),h=d("p"),h.innerHTML=re,z=o(),B=d("p"),B.innerHTML=le,L=o(),v(J.$$.fragment),H=o(),v(G.$$.fragment),O=o(),f=d("div"),v(j.$$.fragment),K=o(),X=d("p"),X.textContent=pe,ee=o(),E=d("p"),E.innerHTML=de,ne=o(),$=d("div"),v(I.$$.fragment),te=o(),N=d("p"),N.textContent=ue,se=o(),v(x.$$.fragment),Y=o(),v(Z.$$.fragment),R=o(),w=d("div"),v(C.$$.fragment),ie=o(),U=d("p"),U.textContent=fe,Q=o(),W=d("p"),this.h()},l(e){const n=$e("svelte-u9bgzb",document.head);s=u(n,"META",{name:!0,content:!0}),n.forEach(t),g=r(e),l=u(e,"P",{}),A(l).forEach(t),p=r(e),y(c.$$.fragment,e),i=r(e),h=u(e,"P",{"data-svelte-h":!0}),k(h)!=="svelte-z4ffbo"&&(h.innerHTML=re),z=r(e),B=u(e,"P",{"data-svelte-h":!0}),k(B)!=="svelte-dzyais"&&(B.innerHTML=le),L=r(e),y(J.$$.fragment,e),H=r(e),y(G.$$.fragment,e),O=r(e),f=u(e,"DIV",{class:!0});var _=A(f);y(j.$$.fragment,_),K=r(_),X=u(_,"P",{"data-svelte-h":!0}),k(X)!=="svelte-1jvczvp"&&(X.textContent=pe),ee=r(_),E=u(_,"P",{"data-svelte-h":!0}),k(E)!=="svelte-496sm0"&&(E.innerHTML=de),ne=r(_),$=u(_,"DIV",{class:!0});var b=A($);y(I.$$.fragment,b),te=r(b),N=u(b,"P",{"data-svelte-h":!0}),k(N)!=="svelte-50j04k"&&(N.textContent=ue),se=r(b),y(x.$$.fragment,b),b.forEach(t),_.forEach(t),Y=r(e),y(Z.$$.fragment,e),R=r(e),w=u(e,"DIV",{class:!0});var F=A(w);y(C.$$.fragment,F),ie=r(F),U=u(F,"P",{"data-svelte-h":!0}),k(U)!=="svelte-19ryw33"&&(U.textContent=fe),F.forEach(t),Q=r(e),W=u(e,"P",{}),A(W).forEach(t),this.h()},h(){S(s,"name","hf:doc:metadata"),S(s,"content",Pe),S($,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),S(f,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),S(w,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8")},m(e,n){m(document.head,s),a(e,g,n),a(e,l,n),a(e,p,n),T(c,e,n),a(e,i,n),a(e,h,n),a(e,z,n),a(e,B,n),a(e,L,n),T(J,e,n),a(e,H,n),T(G,e,n),a(e,O,n),a(e,f,n),T(j,f,null),m(f,K),m(f,X),m(f,ee),m(f,E),m(f,ne),m(f,$),T(I,$,null),m($,te),m($,N),m($,se),T(x,$,null),a(e,Y,n),T(Z,e,n),a(e,R,n),a(e,w,n),T(C,w,null),m(w,ie),m(w,U),a(e,Q,n),a(e,W,n),q=!0},p(e,[n]){const _={};n&2&&(_.$$scope={dirty:n,ctx:e}),J.$set(_);const b={};n&2&&(b.$$scope={dirty:n,ctx:e}),x.$set(b)},i(e){q||(D(c.$$.fragment,e),D(J.$$.fragment,e),D(G.$$.fragment,e),D(j.$$.fragment,e),D(I.$$.fragment,e),D(x.$$.fragment,e),D(Z.$$.fragment,e),D(C.$$.fragment,e),q=!0)},o(e){M(c.$$.fragment,e),M(J.$$.fragment,e),M(G.$$.fragment,e),M(j.$$.fragment,e),M(I.$$.fragment,e),M(x.$$.fragment,e),M(Z.$$.fragment,e),M(C.$$.fragment,e),q=!1},d(e){e&&(t(g),t(l),t(p),t(i),t(h),t(z),t(B),t(L),t(H),t(O),t(f),t(Y),t(R),t(w),t(Q),t(W)),t(s),P(c,e),P(J,e),P(G,e),P(j),P(I),P(x),P(Z,e),P(C)}}}const Pe='{"title":"Dance Diffusion","local":"dance-diffusion","sections":[{"title":"DanceDiffusionPipeline","local":"diffusers.DanceDiffusionPipeline","sections":[],"depth":2},{"title":"AudioPipelineOutput","local":"diffusers.AudioPipelineOutput","sections":[],"depth":2}],"depth":1}';function Je(V){return he(()=>{new URLSearchParams(window.location.search).get("fw"),window.hfDocGoto=we}),[]}class Ce extends _e{constructor(s){super(),ge(this,s,Je,Me,me,{})}}export{Ce as component};
