o pi=@sddlmZmZmZddlmZmZddlmZGdddeZ GdddZ Gd d d e Z Gd d d e Z Gd dde Z dS))AnyDictList) ConfigMixinregister_to_config) CONFIG_NAMEcsleZdZdZeZed fdd Zede e fddZ de e e ffd d Zde e e ffd d ZZS)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. ?Ncspt|dur |dus|dur|durtd|dur4t|tr0d|kr/dks6tdtddSdS)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__Q/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/callbacks.pyr s zPipelineCallback.__init__returncCtd|j)Nz2You need to set the attribute `tensor_inputs` for NotImplementedErrorrrrrr tensor_inputs#zPipelineCallback.tensor_inputscCr)Nz3You need to implement the method `callback_fn` for r)rpipeline step_index timestepscallback_kwargsrrr callback_fn'zPipelineCallback.callback_fncCs|||||SN)r#)rrr timestepr"rrr__call__*r$zPipelineCallback.__call__)r N)__name__ __module__ __qualname____doc__r config_namerr propertyrstrrrrr#r' __classcell__rrrrr s  r c@sNeZdZdZdeefddZedeefddZ de ee ffdd Z d S) 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. callbackscCs ||_dSr%r1)rr1rrrr 4s zMultiPipelineCallbacks.__init__rcCsdd|jDS)NcSsg|] }|jD]}|qqSr)r).0callbackinputrrr 9sz8MultiPipelineCallbacks.tensor_inputs..r2rrrrr7rz$MultiPipelineCallbacks.tensor_inputscCs|jD] }|||||}q|S)zr Calls all the callbacks in order with the given arguments and returns the final callback_kwargs. r2)rrr r&r"r4rrrr';s zMultiPipelineCallbacks.__call__N) r(r)r*r+rr r r-r.rrrr'rrrrr0.s r0c@s,eZdZdZdgZdeeeffddZdS)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 Csd|jj}|jj}|dur|nt|j|}||kr0||jd}|dd}d|_|||jd<|S)Nrr configrrint num_timestepsr_guidance_scale) rrr r&r"rr cutoff_stepr8rrrr#Os zSDCFGCutoffCallback.callback_fnN r(r)r*r+rrr.rr#rrrrr7Esr7c@s.eZdZdZgdZdeeeffddZdS)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. )r8add_text_embeds add_time_idsrc Cs|jj}|jj}|dur|nt|j|}||krX||jd}|dd}||jd} | dd} ||jd} | dd} d|_|||jd<| ||jd<| ||jd<|S)Nrr9rr r:) rrr r&r"rrr?r8rBrCrrrr#ls    z!SDXLCFGCutoffCallback.callback_fnNr@rrrrrAbsrAc@s*eZdZdZgZdeeeffddZdS)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. rcCs@|jj}|jj}|dur|nt|j|}||kr|d|S)Nr )r;rrr<r=set_ip_adapter_scale)rrr r&r"rrr?rrrr#s z(IPAdapterScaleCutoffCallback.callback_fnNr@rrrrrEsrEN)typingrrrconfiguration_utilsrrutilsrr r0r7rArErrrrs '%