wi5(ddlZddlZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZe jeZe r dd lmZdd lmZd eieed d dideeddeededddZ dZ!GddZ"dS)N) nullcontext)Optional)validate_hf_hub_args) deprecateis_accelerate_availablelogging) SingleFileComponentErrorconvert_controlnet_checkpointconvert_ldm_unet_checkpointconvert_ldm_vae_checkpoint/convert_sd3_transformer_checkpoint_to_diffusers4convert_stable_cascade_unet_single_file_to_diffusers+create_controlnet_diffusers_config_from_ldm%create_unet_diffusers_config_from_ldm$create_vae_diffusers_config_from_ldmfetch_diffusers_configfetch_original_configload_single_file_checkpoint)init_empty_weights)load_model_dict_into_metacheckpoint_mapping_fnunetnum_in_channels in_channels)rconfig_mapping_fndefault_subfolder legacy_kwargsvae)rrr)rr transformer)rr)StableCascadeUNetUNet2DConditionModel AutoencoderKLControlNetModelSD3Transformer2DModelc dtj|j}i}|D]}||vr ||||<|SN)inspect signature parameters) mapping_fnkwargsr+mapping_kwargs parameters s/root/.openclaw/workspace/chatterbox_venv_py311/lib/python3.11/site-packages/diffusers/loaders/single_file_model.py_get_mapping_function_kwargsr1KsJ":..9JN::   (.y(9N9 % cLeZdZdZeeddeefdZdS)FromOriginalModelMixinz] Load pretrained weights saved in the `.ckpt` or `.safetensors` format into a model. N%pretrained_model_link_or_path_or_dictc !"#|j}|tvr|d}|D]!\}}|vr||<"!"fd D}||t|fi}|d)||d!|}|st'd"|d#t)rt*nt,}|5||}dddn #1swxYwYt)r~t1|||$} |j|jD]##fd%| D} t5| d&kr9t6d'|jd(d| gn|||||||S)*a Instantiate a model from pretrained weights saved in the original `.ckpt` or `.safetensors` format. The model is set in evaluation mode (`model.eval()`) by default. Parameters: pretrained_model_link_or_path_or_dict (`str`, *optional*): Can be either: - A link to the `.safetensors` or `.ckpt` file (for example `"https://huggingface.co//blob/main/.safetensors"`) on the Hub. - A path to a local *file* containing the weights of the component model. - A state dict containing the component model weights. config (`str`, *optional*): - A string, the *repo id* (for example `CompVis/ldm-text2im-large-256`) of a pretrained pipeline hosted on the Hub. - A path to a *directory* (for example `./my_pipeline_directory/`) containing the pipeline component configs in Diffusers format. subfolder (`str`, *optional*, defaults to `""`): The subfolder location of a model file within a larger model repository on the Hub or locally. original_config (`str`, *optional*): Dict or path to a yaml file containing the configuration for the model in its original format. If a dict is provided, it will be used to initialize the model configuration. torch_dtype (`str` or `torch.dtype`, *optional*): Override the default `torch.dtype` and load the model with another dtype. If `"auto"` is passed, the dtype is automatically derived from the model's weights. force_download (`bool`, *optional*, defaults to `False`): Whether or not to force the (re-)download of the model weights and configuration files, overriding the cached versions if they exist. cache_dir (`Union[str, os.PathLike]`, *optional*): Path to a directory where a downloaded pretrained model configuration is cached if the standard cache is not used. resume_download (`bool`, *optional*, defaults to `False`): Whether or not to resume downloading the model weights and configuration files. If set to `False`, any incompletely downloaded files are deleted. proxies (`Dict[str, str]`, *optional*): A dictionary of proxy servers to use by protocol or endpoint, for example, `{'http': 'foo.bar:3128', 'http://hostname': 'foo.bar:4012'}`. The proxies are used on each request. local_files_only (`bool`, *optional*, defaults to `False`): Whether to only load local model weights and configuration files or not. If set to True, the model won't be downloaded from the Hub. token (`str` or *bool*, *optional*): The token to use as HTTP bearer authorization for remote files. If `True`, the token generated from `diffusers-cli login` (stored in `~/.huggingface`) is used. revision (`str`, *optional*, defaults to `"main"`): The specific model version to use. It can be a branch name, a tag name, a commit id, or any identifier allowed by Git. kwargs (remaining dictionary of keyword arguments, *optional*): Can be used to overwrite load and saveable variables (for example the pipeline components of the specific pipeline class). The overwritten components are directly passed to the pipelines `__init__` method. See example below for more information. ```py >>> from diffusers import StableCascadeUNet >>> ckpt_path = "https://huggingface.co/stabilityai/stable-cascade/blob/main/stage_b_lite.safetensors" >>> model = StableCascadeUNet.from_single_file(ckpt_path) ``` z9FromOriginalModelMixin is currently only compatible with z, pretrained_model_link_or_pathNzUPlease use `pretrained_model_link_or_path_or_dict` argument instead for model classesz1.0.0configoriginal_configzz`from_single_file` cannot accept both `config` and `original_config` arguments. Please provide only one of these argumentsresume_downloadforce_downloadFproxiestoken cache_dirlocal_files_only subfolderrevision torch_dtype)r:r;r<r=r>r?rArrz(`original_config` has been provided for z~ but no mapping functionwas found to convert the original config to a Diffusers config in`diffusers.loaders.single_file_utils`)r?)r9 checkpointzqInvalid `config` argument. Please provide a string representing a repo idor path to a local Diffusers model repo.pretrained_model_name_or_pathr)rDr@r?rcLi|] }|vs|v ||!S)get).0kexpected_kwargsr-optional_kwargss r0 z;FromOriginalModelMixin.from_single_file..s;mmm_@T@TXY]lXlXlAvzz!}}XlXlXlr2)r8rCzFailed to load zD. Weights for this component appear to be missing in the checkpoint.)dtypec>g|]}tj||Sr()research)rHrIpats r0 z;FromOriginalModelMixin.from_single_file..s+&_&_&_QRYsTUEVEVE^qE^E^E^r2rzESome weights of the model checkpoint were not used when initializing z: rF) __name__SINGLE_FILE_LOADABLE_CLASSES ValueErrorjoinkeysrGrpop isinstancedictrstrrr1r load_config_get_signature_keysitemsupdater rrr from_configr"_keys_to_ignore_on_load_unexpectedlenloggerwarningload_state_dicttoeval)$clsr5r- class_namer7deprecation_messager8r9r:r;r<r=r>r?r@rArBrCmapping_functionsrrconfig_mapping_kwargsdiffusers_model_config$default_pretrained_model_config_namer legacy_keynew_key model_kwargscheckpoint_mapping_kwargsdiffusers_format_checkpointctxmodelunexpected_keysrJrKrQs$ ` @@@r0from_single_filez'FromOriginalModelMixin.from_single_file[sz\ 9 9 9|DIIVrVwVwVyVyLzLz|| )/ 3RTX(Y(Y% ( 4g  5w@S T T T4Q 1Hd++ **%6==  /"=M !**%6==$4e<<**Y-- 7D))JJ{D11 !::&8$??JJ{D11 ::j$//jj55 ;T B B >JJ45 /-#!1!   J9D 12I J = 8"&777$56I$J!!$(! ( @:@@@/3// l"7Zj"k"k"k$@AR$]$]V\$]$] !%6%6& /J&&J_&& " " fc**;A88$G0 ;;7=>]7^4&*;;; 12E FI%** &)__.R#!1&5&& " 03/F/Fs/K/K ,O_"333 1/ B +8+>+>+@+@AA'J!V++*0**Z*@*@wmmmmmmfmmmL " ) ), 7 7 7$@AV$a$aZ`$a$a!&;&;' )j' ' D]' ' #+ *r*rrr %<$=$=N  ; SUU < <OO$:;;E < < < < < < < < < < < < < < < # $ $ ?7?ZbmnnnO7C C``C&_&_&_&_/&_&_&_OO?##a''N\_\hNNptpypy{JqKqKpLNN  ! !"= > > >  " HH[ ! ! !  s:NN #N r() rS __module__ __qualname____doc__ classmethodrrr[rwrFr2r0r4r4VsaJJXc]JJJ[JJJr2r4)#r)rO contextlibrtypingrhuggingface_hub.utilsrutilsrrr single_file_utilsr r r rrrrrrrrr get_loggerrSrc acceleratermodels.modeling_utilsrrTr1r4rFr2r0rs """"""666666??????????                              H % %B------AAAAAA !U"=B# } "<A" "?H "Q*+  8QQQQQQQQQQr2