import{s as Ee,n as Oe,o as He}from"../chunks/scheduler.53228c21.js";import{S as Ie,i as qe,e as a,s as r,c as p,h as ze,a as c,d as t,b as o,f as y,g as u,j as v,k as M,l as n,m as d,n as m,t as f,o as h,p as g}from"../chunks/index.100fac89.js";import{C as Ne}from"../chunks/CopyLLMTxtMenu.9360b11d.js";import{D as W}from"../chunks/Docstring.22cc8849.js";import{H as Se,E as Fe}from"../chunks/MermaidChart.svelte_svelte_type_style_lang.c9001d7a.js";function Ue(ye){let _,Y,R,J,P,Q,T,X,A,Me='The <code>KDPM2DiscreteScheduler</code> with ancestral sampling is inspired by the <a href="https://huggingface.co/papers/2206.00364" rel="nofollow">Elucidating the Design Space of Diffusion-Based Generative Models</a> paper, and the scheduler is ported from and created by <a href="https://github.com/crowsonkb/" rel="nofollow">Katherine Crowson</a>.',Z,K,Pe='The original codebase can be found at <a href="https://github.com/crowsonkb/k-diffusion" rel="nofollow">crowsonkb/k-diffusion</a>.',ee,w,te,i,C,ue,z,Te=`KDPM2DiscreteScheduler with ancestral sampling is inspired by the DPMSolver2 and Algorithm 2 from the <a href="https://huggingface.co/papers/2206.00364" rel="nofollow">Elucidating
the Design Space of Diffusion-Based Generative Models</a> paper.`,me,N,Ae=`This model inherits from <a href="/docs/diffusers/main/en/api/schedulers/overview#diffusers.SchedulerMixin">SchedulerMixin</a> and <a href="/docs/diffusers/main/en/api/configuration#diffusers.ConfigMixin">ConfigMixin</a>. Check the superclass documentation for the generic
methods the library implements for all schedulers such as loading and saving.`,fe,D,k,he,F,Ke=`Ensures interchangeability with schedulers that need to scale the denoising model input depending on the
current timestep.`,ge,$,L,_e,U,we="Sets the begin index for the scheduler. This function should be run from pipeline before the inference.",be,x,E,ve,V,Ce="Sets the discrete timesteps used for the diffusion chain (to be run before inference).",De,S,O,$e,G,ke=`Predict the sample from the previous timestep by reversing the SDE. This function propagates the diffusion
process from the learned model outputs (most often the predicted noise).`,se,H,ne,b,I,xe,B,Le="Base class for the output of a scheduler’s <code>step</code> function.",re,q,oe,j,ie;return P=new Ne({props:{containerStyle:"float: right; margin-left: 10px; display: inline-flex; position: relative; z-index: 10;"}}),T=new Se({props:{title:"KDPM2AncestralDiscreteScheduler",local:"kdpm2ancestraldiscretescheduler",headingTag:"h1"}}),w=new Se({props:{title:"KDPM2AncestralDiscreteScheduler",local:"diffusers.KDPM2AncestralDiscreteScheduler",headingTag:"h2"}}),C=new W({props:{name:"class diffusers.KDPM2AncestralDiscreteScheduler",anchor:"diffusers.KDPM2AncestralDiscreteScheduler",parameters:[{name:"num_train_timesteps",val:": int = 1000"},{name:"beta_start",val:": float = 0.00085"},{name:"beta_end",val:": float = 0.012"},{name:"beta_schedule",val:": str = 'linear'"},{name:"trained_betas",val:": typing.Union[numpy.ndarray, typing.List[float], NoneType] = None"},{name:"use_karras_sigmas",val:": typing.Optional[bool] = False"},{name:"use_exponential_sigmas",val:": typing.Optional[bool] = False"},{name:"use_beta_sigmas",val:": typing.Optional[bool] = False"},{name:"prediction_type",val:": str = 'epsilon'"},{name:"timestep_spacing",val:": str = 'linspace'"},{name:"steps_offset",val:": int = 0"}],parametersDescription:[{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.num_train_timesteps",description:`<strong>num_train_timesteps</strong> (<code>int</code>, defaults to 1000) &#x2014;
The number of diffusion steps to train the model.`,name:"num_train_timesteps"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.beta_start",description:`<strong>beta_start</strong> (<code>float</code>, defaults to 0.00085) &#x2014;
The starting <code>beta</code> value of inference.`,name:"beta_start"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.beta_end",description:`<strong>beta_end</strong> (<code>float</code>, defaults to 0.012) &#x2014;
The final <code>beta</code> value.`,name:"beta_end"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.beta_schedule",description:`<strong>beta_schedule</strong> (<code>str</code>, defaults to <code>&quot;linear&quot;</code>) &#x2014;
The beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
<code>linear</code> or <code>scaled_linear</code>.`,name:"beta_schedule"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.trained_betas",description:`<strong>trained_betas</strong> (<code>np.ndarray</code>, <em>optional</em>) &#x2014;
Pass an array of betas directly to the constructor to bypass <code>beta_start</code> and <code>beta_end</code>.`,name:"trained_betas"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.use_karras_sigmas",description:`<strong>use_karras_sigmas</strong> (<code>bool</code>, <em>optional</em>, defaults to <code>False</code>) &#x2014;
Whether to use Karras sigmas for step sizes in the noise schedule during the sampling process. If <code>True</code>,
the sigmas are determined according to a sequence of noise levels {&#x3C3;i}.`,name:"use_karras_sigmas"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.use_exponential_sigmas",description:`<strong>use_exponential_sigmas</strong> (<code>bool</code>, <em>optional</em>, defaults to <code>False</code>) &#x2014;
Whether to use exponential sigmas for step sizes in the noise schedule during the sampling process.`,name:"use_exponential_sigmas"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.use_beta_sigmas",description:`<strong>use_beta_sigmas</strong> (<code>bool</code>, <em>optional</em>, defaults to <code>False</code>) &#x2014;
Whether to use beta sigmas for step sizes in the noise schedule during the sampling process. Refer to <a href="https://huggingface.co/papers/2407.12173" rel="nofollow">Beta
Sampling is All You Need</a> for more information.`,name:"use_beta_sigmas"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.prediction_type",description:`<strong>prediction_type</strong> (<code>str</code>, defaults to <code>epsilon</code>, <em>optional</em>) &#x2014;
Prediction type of the scheduler function; can be <code>epsilon</code> (predicts the noise of the diffusion process),
<code>sample</code> (directly predicts the noisy sample<code>) or </code>v_prediction\` (see section 2.4 of <a href="https://imagen.research.google/video/paper.pdf" rel="nofollow">Imagen
Video</a> paper).`,name:"prediction_type"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.timestep_spacing",description:`<strong>timestep_spacing</strong> (<code>str</code>, defaults to <code>&quot;linspace&quot;</code>) &#x2014;
The way the timesteps should be scaled. Refer to Table 2 of the <a href="https://huggingface.co/papers/2305.08891" rel="nofollow">Common Diffusion Noise Schedules and
Sample Steps are Flawed</a> for more information.`,name:"timestep_spacing"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.steps_offset",description:`<strong>steps_offset</strong> (<code>int</code>, defaults to 0) &#x2014;
An offset added to the inference steps, as required by some model families.`,name:"steps_offset"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py#L96"}}),k=new W({props:{name:"scale_model_input",anchor:"diffusers.KDPM2AncestralDiscreteScheduler.scale_model_input",parameters:[{name:"sample",val:": Tensor"},{name:"timestep",val:": typing.Union[float, torch.Tensor]"}],parametersDescription:[{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.scale_model_input.sample",description:`<strong>sample</strong> (<code>torch.Tensor</code>) &#x2014;
The input sample.`,name:"sample"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.scale_model_input.timestep",description:`<strong>timestep</strong> (<code>int</code>, <em>optional</em>) &#x2014;
The current timestep in the diffusion chain.`,name:"timestep"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py#L214",returnDescription:`<script context="module">export const metadata = 'undefined';<\/script>


<p>A scaled input sample.</p>
`,returnType:`<script context="module">export const metadata = 'undefined';<\/script>


<p><code>torch.Tensor</code></p>
`}}),L=new W({props:{name:"set_begin_index",anchor:"diffusers.KDPM2AncestralDiscreteScheduler.set_begin_index",parameters:[{name:"begin_index",val:": int = 0"}],parametersDescription:[{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.set_begin_index.begin_index",description:`<strong>begin_index</strong> (<code>int</code>) &#x2014;
The begin index for the scheduler.`,name:"begin_index"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py#L204"}}),E=new W({props:{name:"set_timesteps",anchor:"diffusers.KDPM2AncestralDiscreteScheduler.set_timesteps",parameters:[{name:"num_inference_steps",val:": int"},{name:"device",val:": typing.Union[str, torch.device] = None"},{name:"num_train_timesteps",val:": typing.Optional[int] = None"}],parametersDescription:[{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.set_timesteps.num_inference_steps",description:`<strong>num_inference_steps</strong> (<code>int</code>) &#x2014;
The number of diffusion steps used when generating samples with a pre-trained model.`,name:"num_inference_steps"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.set_timesteps.device",description:`<strong>device</strong> (<code>str</code> or <code>torch.device</code>, <em>optional</em>) &#x2014;
The device to which the timesteps should be moved to. If <code>None</code>, the timesteps are not moved.`,name:"device"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py#L244"}}),O=new W({props:{name:"step",anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step",parameters:[{name:"model_output",val:": typing.Union[torch.Tensor, numpy.ndarray]"},{name:"timestep",val:": typing.Union[float, torch.Tensor]"},{name:"sample",val:": typing.Union[torch.Tensor, numpy.ndarray]"},{name:"generator",val:": typing.Optional[torch._C.Generator] = None"},{name:"return_dict",val:": bool = True"}],parametersDescription:[{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step.model_output",description:`<strong>model_output</strong> (<code>torch.Tensor</code>) &#x2014;
The direct output from learned diffusion model.`,name:"model_output"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step.timestep",description:`<strong>timestep</strong> (<code>float</code>) &#x2014;
The current discrete timestep in the diffusion chain.`,name:"timestep"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step.sample",description:`<strong>sample</strong> (<code>torch.Tensor</code>) &#x2014;
A current instance of a sample created by the diffusion process.`,name:"sample"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step.generator",description:`<strong>generator</strong> (<code>torch.Generator</code>, <em>optional</em>) &#x2014;
A random number generator.`,name:"generator"},{anchor:"diffusers.KDPM2AncestralDiscreteScheduler.step.return_dict",description:`<strong>return_dict</strong> (<code>bool</code>) &#x2014;
Whether or not to return a
<code>KDPM2AncestralDiscreteSchedulerOutput</code> or tuple.`,name:"return_dict"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_k_dpm_2_ancestral_discrete.py#L475",returnDescription:`<script context="module">export const metadata = 'undefined';<\/script>


<p>If return_dict is <code>True</code>,
<code>KDPM2AncestralDiscreteSchedulerOutput</code> is
returned, otherwise a tuple is returned where the first element is the sample tensor.</p>
`,returnType:`<script context="module">export const metadata = 'undefined';<\/script>


<p><code>KDPM2AncestralDiscreteSchedulerOutput</code> or <code>tuple</code></p>
`}}),H=new Se({props:{title:"SchedulerOutput",local:"diffusers.schedulers.scheduling_utils.SchedulerOutput",headingTag:"h2"}}),I=new W({props:{name:"class diffusers.schedulers.scheduling_utils.SchedulerOutput",anchor:"diffusers.schedulers.scheduling_utils.SchedulerOutput",parameters:[{name:"prev_sample",val:": Tensor"}],parametersDescription:[{anchor:"diffusers.schedulers.scheduling_utils.SchedulerOutput.prev_sample",description:`<strong>prev_sample</strong> (<code>torch.Tensor</code> of shape <code>(batch_size, num_channels, height, width)</code> for images) &#x2014;
Computed sample <code>(x_{t-1})</code> of previous timestep. <code>prev_sample</code> should be used as next model input in the
denoising loop.`,name:"prev_sample"}],source:"https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_utils.py#L62"}}),q=new Fe({props:{source:"https://github.com/huggingface/diffusers/blob/main/docs/source/en/api/schedulers/dpm_discrete_ancestral.md"}}),{c(){_=a("meta"),Y=r(),R=a("p"),J=r(),p(P.$$.fragment),Q=r(),p(T.$$.fragment),X=r(),A=a("p"),A.innerHTML=Me,Z=r(),K=a("p"),K.innerHTML=Pe,ee=r(),p(w.$$.fragment),te=r(),i=a("div"),p(C.$$.fragment),ue=r(),z=a("p"),z.innerHTML=Te,me=r(),N=a("p"),N.innerHTML=Ae,fe=r(),D=a("div"),p(k.$$.fragment),he=r(),F=a("p"),F.textContent=Ke,ge=r(),$=a("div"),p(L.$$.fragment),_e=r(),U=a("p"),U.textContent=we,be=r(),x=a("div"),p(E.$$.fragment),ve=r(),V=a("p"),V.textContent=Ce,De=r(),S=a("div"),p(O.$$.fragment),$e=r(),G=a("p"),G.textContent=ke,se=r(),p(H.$$.fragment),ne=r(),b=a("div"),p(I.$$.fragment),xe=r(),B=a("p"),B.innerHTML=Le,re=r(),p(q.$$.fragment),oe=r(),j=a("p"),this.h()},l(e){const s=ze("svelte-u9bgzb",document.head);_=c(s,"META",{name:!0,content:!0}),s.forEach(t),Y=o(e),R=c(e,"P",{}),y(R).forEach(t),J=o(e),u(P.$$.fragment,e),Q=o(e),u(T.$$.fragment,e),X=o(e),A=c(e,"P",{"data-svelte-h":!0}),v(A)!=="svelte-1xsfzct"&&(A.innerHTML=Me),Z=o(e),K=c(e,"P",{"data-svelte-h":!0}),v(K)!=="svelte-loovl7"&&(K.innerHTML=Pe),ee=o(e),u(w.$$.fragment,e),te=o(e),i=c(e,"DIV",{class:!0});var l=y(i);u(C.$$.fragment,l),ue=o(l),z=c(l,"P",{"data-svelte-h":!0}),v(z)!=="svelte-lafslh"&&(z.innerHTML=Te),me=o(l),N=c(l,"P",{"data-svelte-h":!0}),v(N)!=="svelte-linuuh"&&(N.innerHTML=Ae),fe=o(l),D=c(l,"DIV",{class:!0});var ae=y(D);u(k.$$.fragment,ae),he=o(ae),F=c(ae,"P",{"data-svelte-h":!0}),v(F)!=="svelte-1rkfgpx"&&(F.textContent=Ke),ae.forEach(t),ge=o(l),$=c(l,"DIV",{class:!0});var ce=y($);u(L.$$.fragment,ce),_e=o(ce),U=c(ce,"P",{"data-svelte-h":!0}),v(U)!=="svelte-1k141rk"&&(U.textContent=we),ce.forEach(t),be=o(l),x=c(l,"DIV",{class:!0});var de=y(x);u(E.$$.fragment,de),ve=o(de),V=c(de,"P",{"data-svelte-h":!0}),v(V)!=="svelte-1vzm9q"&&(V.textContent=Ce),de.forEach(t),De=o(l),S=c(l,"DIV",{class:!0});var le=y(S);u(O.$$.fragment,le),$e=o(le),G=c(le,"P",{"data-svelte-h":!0}),v(G)!=="svelte-hi84tp"&&(G.textContent=ke),le.forEach(t),l.forEach(t),se=o(e),u(H.$$.fragment,e),ne=o(e),b=c(e,"DIV",{class:!0});var pe=y(b);u(I.$$.fragment,pe),xe=o(pe),B=c(pe,"P",{"data-svelte-h":!0}),v(B)!=="svelte-6ojmkw"&&(B.innerHTML=Le),pe.forEach(t),re=o(e),u(q.$$.fragment,e),oe=o(e),j=c(e,"P",{}),y(j).forEach(t),this.h()},h(){M(_,"name","hf:doc:metadata"),M(_,"content",Ve),M(D,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),M($,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),M(x,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),M(S,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),M(i,"class","docstring border-l-2 border-t-2 pl-4 pt-3.5 border-gray-100 rounded-tl-xl mb-6 mt-8"),M(b,"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,s){n(document.head,_),d(e,Y,s),d(e,R,s),d(e,J,s),m(P,e,s),d(e,Q,s),m(T,e,s),d(e,X,s),d(e,A,s),d(e,Z,s),d(e,K,s),d(e,ee,s),m(w,e,s),d(e,te,s),d(e,i,s),m(C,i,null),n(i,ue),n(i,z),n(i,me),n(i,N),n(i,fe),n(i,D),m(k,D,null),n(D,he),n(D,F),n(i,ge),n(i,$),m(L,$,null),n($,_e),n($,U),n(i,be),n(i,x),m(E,x,null),n(x,ve),n(x,V),n(i,De),n(i,S),m(O,S,null),n(S,$e),n(S,G),d(e,se,s),m(H,e,s),d(e,ne,s),d(e,b,s),m(I,b,null),n(b,xe),n(b,B),d(e,re,s),m(q,e,s),d(e,oe,s),d(e,j,s),ie=!0},p:Oe,i(e){ie||(f(P.$$.fragment,e),f(T.$$.fragment,e),f(w.$$.fragment,e),f(C.$$.fragment,e),f(k.$$.fragment,e),f(L.$$.fragment,e),f(E.$$.fragment,e),f(O.$$.fragment,e),f(H.$$.fragment,e),f(I.$$.fragment,e),f(q.$$.fragment,e),ie=!0)},o(e){h(P.$$.fragment,e),h(T.$$.fragment,e),h(w.$$.fragment,e),h(C.$$.fragment,e),h(k.$$.fragment,e),h(L.$$.fragment,e),h(E.$$.fragment,e),h(O.$$.fragment,e),h(H.$$.fragment,e),h(I.$$.fragment,e),h(q.$$.fragment,e),ie=!1},d(e){e&&(t(Y),t(R),t(J),t(Q),t(X),t(A),t(Z),t(K),t(ee),t(te),t(i),t(se),t(ne),t(b),t(re),t(oe),t(j)),t(_),g(P,e),g(T,e),g(w,e),g(C),g(k),g(L),g(E),g(O),g(H,e),g(I),g(q,e)}}}const Ve='{"title":"KDPM2AncestralDiscreteScheduler","local":"kdpm2ancestraldiscretescheduler","sections":[{"title":"KDPM2AncestralDiscreteScheduler","local":"diffusers.KDPM2AncestralDiscreteScheduler","sections":[],"depth":2},{"title":"SchedulerOutput","local":"diffusers.schedulers.scheduling_utils.SchedulerOutput","sections":[],"depth":2}],"depth":1}';function Ge(ye){return He(()=>{new URLSearchParams(window.location.search).get("fw")}),[]}class Je extends Ie{constructor(_){super(),qe(this,_,Ge,Ue,Ee,{})}}export{Je as component};
