wi/DdZddlmZddlmZerddlZdZdS)z7 Accelerate utilities: Utilities related to accelerate )version)is_accelerate_availableNctsStjtjj}tj|tjdkrSfd}|S)a Decorator that applies a registered CpuOffload hook to an arbitrary function rather than `forward`. This is useful for cases where a PyTorch module provides functions other than `forward` that should trigger a move to the appropriate acceleration device. This is the case for `encode` and `decode` in [`AutoencoderKL`]. This decorator looks inside the internal `_hf_hook` property to find a registered offload hook. :param method: The method to decorate. This method should be a method of a PyTorch module. z0.17.0ct|dr/t|jdr|j||g|Ri|S)N_hf_hook pre_forward)hasattrrr )selfargskwargsmethods p/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/utils/accelerate_utils.pywrapperz#apply_forward_hook..wrapper+s^ 4 $ $ , )N)N , M % %d + + +vd,T,,,V,,,)rrparse accelerate __version__ base_version)raccelerate_versionrs` rapply_forward_hookrss # $ $  z'=>>K}'((7=+B+BBB ----- Nr)__doc__ packagingr import_utilsrrrrrrsm111111r