o pi7 @s$ddlZddlZddlZddlmZddlmZddlmZddl m Z m Z m Z ddl mZmZmZmZmZmZmZmZmZmZmZmZmZmZe eZe r\dd lm Z dd l!m"Z"d eieed d dideeddeededdd eid eiedddZ#ddZ$ddZ%GdddZ&dS)N) nullcontext)Optional)validate_hf_hub_args) deprecateis_accelerate_availablelogging)SingleFileComponentError+convert_animatediff_checkpoint_to_diffusersconvert_controlnet_checkpoint0convert_flux_transformer_checkpoint_to_diffusersconvert_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 AutoencoderKLControlNetModelSD3Transformer2DModel MotionAdapterSparseControlNetModelFluxTransformer2DModelcCs>ttdd}tD]}t||}t||r|Sq dS)N.r) importlib import_module__name__splitSINGLE_FILE_LOADABLE_CLASSESgetattr issubclass)clsdiffusers_moduleloadable_class_strloadable_classr7a/home/ubuntu/SoloSpeech/.venv/lib/python3.10/site-packages/diffusers/loaders/single_file_model.py'_get_single_file_loadable_mapping_classXs  r9cKs2t|j}i}|D] }||vr||||<q |SN)inspect signature parameters) mapping_fnkwargsr=mapping_kwargs parameterr7r7r8_get_mapping_function_kwargscs  rBc@s,eZdZdZeeddeefddZdS)FromOriginalModelMixinz] Load pretrained weights saved in the `.ckpt` or `.safetensors` format into a model. N%pretrained_model_link_or_path_or_dictc! slt|}|durtddtdd}|dur(d}tdd||}dd}dd}|dur@|dur@td d d }d d} d d} dd} dd} dd} dd}dd}t|t rx|}n t ||| | | | |d}t|}|d}|rd|vr|d}nd}|durtd|dt|t rt || d}t |fi}|d+||d|}nd|rt|t r|}ntdt|}|d}d|vr|d} | p|dd} |j|| | d}||\d|vr|d}|D]\}}|vr||<qfdd D}||t |fi}|d+||d!|}|sCtd"|d#trItnt}| ||}Wdn 1s_wYtrpt|||d$}n |j|d d%\} }|jdur|jD] fd&d'|D}qt|d(krtd)|jd*d|g|dur||| |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. 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) ``` Nz9FromOriginalModelMixin is currently only compatible with z, pretrained_model_link_or_pathzUPlease 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 argumentsforce_downloadFproxiestoken cache_dirlocal_files_only subfolderrevision torch_dtype)rHrIrJrKrLrNrrz(`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`)rL)rG checkpointzqInvalid `config` argument. Please provide a string representing a repo idor path to a local Diffusers model repo.pretrained_model_name_or_pathr)rQrMrLr cs(i|]}|vs |vr||qSr7)get.0k)expected_kwargsr?optional_kwargsr7r8 s(z;FromOriginalModelMixin.from_single_file..)rFrPzFailed to load zD. Weights for this component appear to be missing in the checkpoint.)dtype)strictcs g|] }t|dur|qSr:)researchrS)patr7r8 2s z;FromOriginalModelMixin.from_single_file..rzESome weights of the model checkpoint were not used when initializing z: r7)!r9 ValueErrorjoinr0keysrRrpop isinstancedictrstrrrBr load_config_get_signature_keysitemsupdater rrr from_configrload_state_dict"_keys_to_ignore_on_load_unexpectedlenloggerwarningr.toeval)!r3rDr?mapping_class_namerEdeprecation_messagerFrGrHrIrJrKrLrMrNrOrPmapping_functionsrrconfig_mapping_kwargsdiffusers_model_config$default_pretrained_model_config_namer legacy_keynew_key model_kwargscheckpoint_mapping_kwargsdiffusers_format_checkpointctxmodelunexpected_keys_r7)rVr?rWr]r8from_single_filess;                            z'FromOriginalModelMixin.from_single_filer:) r. __module__ __qualname____doc__ classmethodrrrerr7r7r7r8rCns rC)'r,r;r[ contextlibrtypingrhuggingface_hub.utilsrutilsrrrsingle_file_utilsr r r r rrrrrrrrrr get_loggerr.rn acceleratermodels.modeling_utilsrr0r9rBrCr7r7r7r8sR    @    &