o p’×i/ã@s6dZddlmZddlmZeƒrddlZdd„ZdS)z7 Accelerate utilities: Utilities related to accelerate é)Úversioné)Úis_accelerate_availableNcs@tƒsˆSt tj¡j}t |¡t d¡krˆS‡fdd„}|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.0cs8t|dƒrt|jdƒr|j |¡ˆ|g|¢Ri|¤ŽS)NÚ_hf_hookÚ pre_forward)Úhasattrrr)ÚselfÚargsÚkwargs©Úmethod©ú^/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/utils/accelerate_utils.pyÚwrapper+s z#apply_forward_hook..wrapper)rrÚparseÚ accelerateÚ __version__Ú base_version)r Úaccelerate_versionrr r rÚapply_forward_hooks  r)Ú__doc__Ú packagingrÚ import_utilsrrrr r r rÚs