+h*(SSKJrJrJr SSKJrJr SSKJr "SS\5r "SS5r "S S \ 5r "S S \ 5r "S S\ 5r "SS\ 5r"SS\ 5rg))AnyDictList) ConfigMixinregister_to_config) CONFIG_NAMEc^\rSrSrSr\r\S U4Sjj5r\ S\ \ 4Sj5r S\ \ \44SjrS\ \ \44SjrSrU=r$) PipelineCallbacka  Base class for all the official callbacks used in a pipeline. This class provides a structure for implementing custom callbacks and ensures that all callbacks have a consistent interface. Please implement the following: `tensor_inputs`: This should return a list of tensor inputs specific to your callback. You will only be able to include variables listed in the `._callback_tensor_inputs` attribute of your pipeline class. `callback_fn`: This method defines the core functionality of your callback. c>[TU]5 UcUbUbUb [S5eUb0[U[5(aSUs=::aS::d O [S5egg)NzSEither cutoff_step_ratio or cutoff_step_index should be provided, not both or none.?z6cutoff_step_ratio must be a float between 0.0 and 1.0.)super__init__ ValueError isinstancefloat)selfcutoff_step_ratiocutoff_step_index __class__s M/home/james-whalen/.local/lib/python3.13/site-packages/diffusers/callbacks.pyrPipelineCallback.__init__sk   %*;*C  ).?.Krs s  (,e44SDU=\Y\=\UV V>] )returnc2[SUR35e)Nz2You need to set the attribute `tensor_inputs` for NotImplementedErrorr)rs r tensor_inputsPipelineCallback.tensor_inputs#s!$VW[WeWeVf"ghhrc2[SUR35e)Nz3You need to implement the method `callback_fn` for r)rpipeline step_index timestepscallback_kwargss r callback_fnPipelineCallback.callback_fn's!$WX\XfXfWg"hiirc&URXX45$N)r')rr#r$timestepr&s r__call__PipelineCallback.__call__*shPPr)rN)__name__ __module__ __qualname____firstlineno____doc__r config_namerrpropertyrstrr rrr'r,__static_attributes__ __classcell__)rs@rr r s} K W WitCyiijtTWY\T\~jQ4PSUXPX>QQrr cb\rSrSrSrS\\4Sjr\S\\ 4Sj5r S\ \ \ 44Sjr Srg ) MultiPipelineCallbacks.z This class is designed to handle multiple pipeline callbacks. It accepts a list of PipelineCallback objects and provides a unified interface for calling all of them. callbackscXlgr*r<)rr<s rrMultiPipelineCallbacks.__init__4s"rrcpURVVs/sHoRHo"PM M snn$s snnfr*)r<r )rcallbackinputs rr $MultiPipelineCallbacks.tensor_inputs7s+&*nnYn(BXBXBXnYYYs2c>URH nU"XX45nM U$)zb Calls all the callbacks in order with the given arguments and returns the final callback_kwargs. r>)rr#r$r+r&rAs rr,MultiPipelineCallbacks.__call__;s&H&xXWO'rr>N)r/r0r1r2r3rr rr5r6r rrr,r7r.rrr:r:.sR #$'7"8#ZtCyZZ4PSUXPX>rr:c6\rSrSrSrS/rS\\\44Sjr Sr g)SDCFGCutoffCallbackEa Callback function for Stable Diffusion Pipelines. After certain number of steps (set by `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. prompt_embedsrcURRnURRnUbUO[URU-5nX':Xa.X@R SnUSSnSUlXUR S'U$)Nrrconfigrrint num_timestepsr _guidance_scale) rr#r$r+r&rr cutoff_steprIs rr'SDCFGCutoffCallback.callback_fnOs KK99 KK99"3!> CH^H^arHrDs   $+,>,>q,ABM)"#.M'*H $5BD..q1 2rr.N r/r0r1r2r3r rr6rr'r7r.rrrGrGEs(%%MdSVX[S[nrrGc8\rSrSrSr/SQrS\\\44Sjr Sr g)SDXLCFGCutoffCallbackba* Callback function for the base Stable Diffusion XL Pipelines. After certain number of steps (set by `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. )rIadd_text_embeds add_time_idsrcURRnURRnUbUO[URU-5nX':Xa|X@R SnUSSnX@R Sn U SSn X@R Sn U SSn SUlXUR S'XUR S'XUR S'U$)NrrKrrrL) rr#r$r+r&rrrQrIrWrXs rr'!SDXLCFGCutoffCallback.callback_fnps KK99 KK99"3!> CH^H^arHrDs   $+,>,>q,ABM)"#.M-.@.@.CDO-bc2O*+=+=a+@AL',L'*H $5BD..q1 25DD..q1 25AD..q1 2rr.NrSr.rrrUrUbs%M dSVX[S[nrrUc8\rSrSrSr/SQrS\\\44Sjr Sr g)SDXLControlnetCFGCutoffCallbacka0 Callback function for the Controlnet Stable Diffusion XL Pipelines. After certain number of steps (set by `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. )rIrWrXimagercURRnURRnUbUO[URU-5nX':XaX@R SnUSSnX@R Sn U SSn X@R Sn U SSn X@R Sn U SSn SUlXUR S'XUR S'XUR S'XUR S'U$)NrrKrrZrrL) rr#r$r+r&rrrQrIrWrXr_s rr'+SDXLControlnetCFGCutoffCallback.callback_fns$ KK99 KK99"3!> CH^H^arHrDs   $+,>,>q,ABM)"#.M-.@.@.CDO-bc2O*+=+=a+@AL',L$$6$6q$9:E"#JE'*H $5BD..q1 25DD..q1 25AD..q1 25:D..q1 2rr.NrSr.rrr]r]s%MdSVX[S[nrr]c4\rSrSrSr/rS\\\44Sjr Sr g)IPAdapterScaleCutoffCallbackaA Callback function for any pipeline that inherits `IPAdapterMixin`. After certain number of steps (set by `cutoff_step_ratio` or `cutoff_step_index`), this callback will set the IP Adapter scale to `0.0`. Note: This callback mutates the IP Adapter attention processors by setting the scale to 0.0 after the cutoff step. rcURRnURRnUbUO[URU-5nX':XaUR S5 U$)Nr)rMrrrNrOset_ip_adapter_scale)rr#r$r+r&rrrQs rr'(IPAdapterScaleCutoffCallback.callback_fnsa KK99 KK99"3!> CH^H^arHrDs   $  ) )# .rr.NrSr.rrrdrds%M dSVX[S[n rrdc8\rSrSrSrSS/rS\\\44Sjr Sr g) SD3CFGCutoffCallbacka  Callback function for Stable Diffusion 3 Pipelines. After certain number of steps (set by `cutoff_step_ratio` or `cutoff_step_index`), this callback will disable the CFG. Note: This callback mutates the pipeline by changing the `_guidance_scale` attribute to 0.0 after the cutoff step. rIpooled_prompt_embedsrcLURRnURRnUbUO[URU-5nX':XaUX@R SnUSSnX@R Sn U SSn SUlXUR S'XUR S'U$)NrrKrrrL) rr#r$r+r&rrrQrIrls rr' SD3CFGCutoffCallback.callback_fns KK99 KK99"3!> CH^H^arHrDs   $+,>,>q,ABM)"#.M#23E3Ea3H#I #7$ (+H $5BD..q1 25ID..q1 2rr.NrSr.rrrjrjs+%&<=MdSVX[S[nrrjN)typingrrrconfiguration_utilsrrutilsr r r:rGrUr]rdrjr.rrrrsm""@$Q{$QN.*:','T-&6-`#30 + r