wi=ddlmZmZmZddlmZmZddlmZGddeZ GddZ Gd d e Z Gd d e Z Gd de Z dS))AnyDictList) ConfigMixinregister_to_config) CONFIG_NAMEceZdZdZeZed fd Zede e fdZ de e e ffdZde e e ffdZxZS) 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. ?Nct||||td|3t|trd|cxkrdksntddSdS)NzSEither cutoff_step_ratio or cutoff_step_index should be provided, not both or none.r z6cutoff_step_ratio must be a float between 0.0 and 1.0.)super__init__ ValueError isinstancefloat)selfcutoff_step_ratiocutoff_step_index __class__s c/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/callbacks.pyrzPipelineCallback.__init__s   %*;*C  ).?.Krss s  (,e44 )=@DU=\=\=\=\Y\=\=\=\=\UVV V ) (=\=\returnc0td|j)Nz2You need to set the attribute `tensor_inputs` for NotImplementedErrorrrs r tensor_inputszPipelineCallback.tensor_inputs#s!"gW[We"g"ghhhrc0td|j)Nz3You need to implement the method `callback_fn` for r)rpipeline step_index timestepscallback_kwargss r callback_fnzPipelineCallback.callback_fn's!"hX\Xf"h"hiiirc2|||||SN)r%)rr!r"timestepr$s r__call__zPipelineCallback.__call__*s*hPPPr)r N)__name__ __module__ __qualname____doc__r config_namerrpropertyrstrrrrr%r) __classcell__)rs@rr r s  K W W W W W WitCyiiiXijtTWY\T\~jjjjQ4PSUXPX>QQQQQQQQrr cneZdZdZdeefdZedeefdZ de ee ffdZ dS)MultiPipelineCallbacksz 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. callbacksc||_dSr'r4)rr4s rrzMultiPipelineCallbacks.__init__4s "rrc$d|jDS)Nc&g|]}|jD]}|S)r).0callbackinputs r z8MultiPipelineCallbacks.tensor_inputs..9s(YYY((BXYYYYYYrr6rs rrz$MultiPipelineCallbacks.tensor_inputs7sYYdnYYYYrc6|jD]}|||||}|S)zr Calls all the callbacks in order with the given arguments and returns the final callback_kwargs. r6)rr!r"r(r$r;s rr)zMultiPipelineCallbacks.__call__;s4 X XH&hxXWWOOrN) r*r+r,r-rr rr/r0rrrr)r9rrr3r3.s #$'7"8####ZtCyZZZXZ4PSUXPX>rr3c4eZdZdZdgZdeeeffdZdS)SDCFGCutoffCallbacka. 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_embedsrc|jj}|jj}||nt|j|z}||kr4||jd}|dd}d|_|||jd<|S)Nrrconfigrrint num_timestepsr_guidance_scale) rr!r"r(r$rr cutoff_steprAs rr%zSDCFGCutoffCallback.callback_fnOs K9 K9"3!>  CH^arHrDsDs   $ $+D,>q,ABM)"##.M'*H $5BOD.q1 2rN r*r+r,r-rrr0rr%r9rrr@r@EsN%%MdSVX[S[nrr@c6eZdZdZgdZdeeeffdZdS)SDXLCFGCutoffCallbacka1 Callback function for 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. )rAadd_text_embeds add_time_idsrc|jj}|jj}||nt|j|z}||kr||jd}|dd}||jd} | dd} ||jd} | dd} d|_|||jd<| ||jd<| ||jd<|S)NrrCrrrD) rr!r"r(r$rrrIrArMrNs rr%z!SDXLCFGCutoffCallback.callback_fnls K9 K9"3!>  CH^arHrDsDs   $ $+D,>q,ABM)"##.M-d.@.CDO-bcc2O*4+=a+@AL',L'*H $5BOD.q1 25DOD.q1 25AOD.q1 2rNrJr9rrrLrLbsRIHHMdSVX[S[nrrLc2eZdZdZgZdeeeffdZdS)IPAdapterScaleCutoffCallbackaQ 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. rc|jj}|jj}||nt|j|z}||kr|d|S)Nr)rErrrFrGset_ip_adapter_scale)rr!r"r(r$rrrIs rr%z(IPAdapterScaleCutoffCallback.callback_fnse K9 K9"3!>  CH^arHrDsDs   $ $  ) )# . . .rNrJr9rrrRrRsKM dSVX[S[n      rrRN)typingrrrconfiguration_utilsrrutilsr r r3r@rLrRr9rrrXs4""""""""""@@@@@@@@$Q$Q$Q$Q$Q{$Q$Q$QN.*:""""","""J#3r