o Gi(@sddlmZddlmZmZddlmZGdddeZGdddZGd d d eZ Gd d d eZ Gd ddeZ GdddeZ GdddeZ dS))Any) 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.rz6cutoff_step_ratio must be a float between 0.0 and 1.0.)super__init__ ValueError isinstancefloat)selfcutoff_step_ratiocutoff_step_index __class__G/home/ubuntu/.local/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!)rrrtimestepr rrr__call__*r"zPipelineCallback.__call__)rN)__name__ __module__ __qualname____doc__r config_namerr propertyliststrrdictrr!r% __classcell__rrrrrs  rc@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)rrrr$r r4rrrr%;s zMultiPipelineCallbacks.__call__N) r&r'r(r)r,rr r+r-rr.rr%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) rrrr$r rr cutoff_stepr8rrrr!Os zSDCFGCutoffCallback.callback_fnN r&r'r(r)rr.r-rr!rrrrr7Esr7c@.eZdZdZgdZdeeeffddZdS)SDXLCFGCutoffCallbacka: 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. )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:) rrrr$r rrr?r8rCrDrrrr!ps    z!SDXLCFGCutoffCallback.callback_fnNr@rrrrrBbsrBc@rA)SDXLControlnetCFGCutoffCallbacka@ 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. )r8rCrDimagerc Cs|jj}|jj}|dur|nt|j|}||krl||jd}|dd}||jd} | dd} ||jd} | dd} ||jd} | dd} d|_|||jd<| ||jd<| ||jd<| ||jd<|S)Nrr9rrEr r:) rrrr$r rrr?r8rCrDrGrrrr!s&    z+SDXLControlnetCFGCutoffCallback.callback_fnNr@rrrrrFsrFc@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)rrrr$r rrr?rrrr!s z(IPAdapterScaleCutoffCallback.callback_fnNr@rrrrrIsrIc@s.eZdZdZddgZdeeeffddZdS)SD3CFGCutoffCallbacka0 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. r8pooled_prompt_embedsrc Cs|jj}|jj}|dur|nt|j|}||krD||jd}|dd}||jd} | dd} d|_|||jd<| ||jd<|S)Nrr9rr r:) rrrr$r rrr?r8rLrrrr!s z SD3CFGCutoffCallback.callback_fnNr@rrrrrKsrKN)typingrconfiguration_utilsrrutilsrrr0r7rBrFrIrKrrrrs  '*0