bi8U ddlZddlZddlZddlmZddlmZddlZddlm Z ddl m Z ddl m Z ddlmZdd lmZmZmZmZdd lmZd d lmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2ejfe4Z5erdd l6m7Z7m8Z8ddl9m:Z:eddr erdZ;ndZ;idde*ide"e.dddidde#e/dddee-dde)d d!d"deid#deid$ed d!d%ed d!d&e$d d!d'e%dd!d(deid)e'd d!d*e!d d!d+ed d!d,e&d d!d-e(d d!e+d d!e+d d!e,dd!e d d!ed d!d.d d!d/Zd2Z?Gd3d4Z@y)5N) nullcontext)Optional)validate_hf_hub_args)Self) __version__)DiffusersAutoQuantizer) deprecateis_accelerate_availableis_torch_versionlogging)empty_device_cache)SingleFileComponentError+convert_animatediff_checkpoint_to_diffusers4convert_auraflow_transformer_checkpoint_to_diffusers.convert_autoencoder_dc_checkpoint_to_diffusers2convert_chroma_transformer_checkpoint_to_diffusersconvert_controlnet_checkpoint2convert_cosmos_transformer_checkpoint_to_diffusers0convert_flux_transformer_checkpoint_to_diffusers(convert_hidream_transformer_to_diffusers.convert_hunyuan_video_transformer_to_diffusersconvert_ldm_unet_checkpointconvert_ldm_vae_checkpoint/convert_ltx_transformer_checkpoint_to_diffusers'convert_ltx_vae_checkpoint_to_diffusersconvert_lumina2_to_diffusers1convert_mochi_transformer_checkpoint_to_diffusers%convert_sana_transformer_to_diffusers/convert_sd3_transformer_checkpoint_to_diffusers4convert_stable_cascade_unet_single_file_to_diffusers$convert_wan_transformer_to_diffusersconvert_wan_vae_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)dispatch_modelinit_empty_weights)load_model_dict_into_metaz>=z1.9.0TFStableCascadeUNetcheckpoint_mapping_fnUNet2DConditionModelunetnum_in_channels in_channels)r/config_mapping_fndefault_subfolder legacy_kwargs AutoencoderKLvae)r/r4r5ControlNetModel)r/r4SD3Transformer2DModel transformer)r/r5 MotionAdapterSparseControlNetModelFluxTransformer2DModelChromaTransformer2DModelLTXVideoTransformer3DModelAutoencoderKLLTXVideo AutoencoderDCMochiTransformer3DModelHunyuanVideoTransformer3DModelAuraFlowTransformer2DModelLumina2Transformer2DModelSanaTransformer2DModelc|SN)xs ^/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/diffusers/loaders/single_file_model.pyrMs1)WanTransformer3DModelWanVACETransformer3DModelAutoencoderKLWanHiDreamImageTransformer2DModelCosmosTransformer3DModelQwenImageTransformer2DModelct|jjt|j SrI)setkeysissubset)model_state_dictcheckpoint_state_dicts rL'_should_convert_state_dict_to_diffusersr[s4#((*+44S9N9S9S9U5VW WWrNctjtjdd}tD]}t ||}t ||s|cSy)N.r) importlib import_module__name__splitSINGLE_FILE_LOADABLE_CLASSESgetattr issubclass)clsdiffusers_moduleloadable_class_strloadable_classs rL'_get_single_file_loadable_mapping_classrisT ..x~~c/B1/EF:& !13EF c> *% % & rNc ptj|j}i}|D]}||vs||||<|SrI)inspect signature parameters) mapping_fnkwargsrmmapping_kwargs parameters rL_get_mapping_function_kwargsrrsK"":.99JN:  (.y(9N9 %: rNc<eZdZdZeeddeedefdZ y)FromOriginalModelMixinz] Load pretrained weights saved in the `.ckpt` or `.safetensors` format into a model. N%pretrained_model_link_or_path_or_dictreturnc t|}|/tddjtj |j dd}|d}t dd||}|jdd}|jdd}| | td |jd d }|jd d} |jd d} |jdd} |jdd} |jdd} |jdd}|jdd}|jdd}|jdd}|jdt}|jdd}|jdd }tddd}||jj|d<|Ct|tjs)tj}t j#d|dt|t$r|}nt'||| | | | ||| }|7t)j*|}|j-|j/|}nd}t|}|d}|Rd |vr|d }nd}|td!|d"t|t0r t3|| #}t5|fi|}|d7||d$|}n|t|t0r|}n:td%t7|}|d&}d'|vr|d'} | xs|jdd} |j9|| | | |(}|j;|\}}d)|vr6|d)} | j=D]\}!}"|!|vs |j|!||"< |D#cic]}#|#|vs|#|vs |#|j |#}$}#|j?|$|rt@ntB}%|%5|j+|}&dddt5|fi|}'tE&jG|r |d7||d*|'}(n|}(|(stId+|d,|jJduxr!|tjLk(xs tO|d-})|)r |jJ}*t|*tPs|*g}*ng}*||jS|&d|(|*.d}+|rq|rtjT|ntjTd/},|&jG}-|(D.cgc] }.|.|-vs|. }/}.d0|,i}+tW|&|(||+||*|/1tYn|&j[|(d 2\}0}/|&j\7|&j\D](}1|/D#cgc]}#t_j`|1|#|#}/}#*tc|/d3kDr5t j#d4|jdd5dj|/g||jg|&||&_4|||&jk||&jm|+d6|+i}2to|&fi|2|&Scc}#w#1swY$xYwcc}.wcc}#w)8au 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 (`torch.dtype`, *optional*): Override the default `torch.dtype` and load the model with another dtype. 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. low_cpu_mem_usage (`bool`, *optional*, defaults to `True` if torch version >= 1.9.0 and is_accelerate_available() else `False`): Speed up model loading only loading the pretrained weights and not initializing the weights. This also tries to not use more than 1x model size in CPU memory (including peak memory) while loading the model. Only supported for PyTorch >= 1.9.0. If you are using an older version of PyTorch, setting this argument to `True` will raise an error. disable_mmap ('bool', *optional*, defaults to 'False'): Whether to disable mmap when loading a Safetensors model. This option can perform better when the model is on a network mount or hard drive, which may not handle the seeky-ness of mmap very well. 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 subfolderrevisionconfig_revision torch_dtypequantization_configlow_cpu_mem_usagedevice disable_mmap single_filepytorch) diffusers file_type frameworkquantzPassed `torch_dtype` z7 is not a `torch.dtype`. Defaulting to `torch.float32`.)r{r|r}r~rrr user_agentr/r4z(`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)rz checkpointzqInvalid `config` argument. Please provide a string representing a repo idor path to a local Diffusers model repo.pretrained_model_name_or_pathr5)rrrr}rr6)ryrzFailed to load zD. Weights for this component appear to be missing in the checkpoint.use_keep_in_fp32_modules)model device_map state_dictkeep_in_fp32_modulescpu)dtyper hf_quantizerrunexpected_keys)strictrzESome weights of the model checkpoint were not used when initializing z: rrJ)8ri ValueErrorjoinrbrWgetr pop_LOW_CPU_MEM_USAGE_DEFAULTr quant_methodvalue isinstancetorchrfloat32loggerwarningdictr*r from_configvalidate_environmentupdate_torch_dtypestrr)rrr( load_config_get_signature_keysitemsupdater,rr[rr_keep_in_fp32_modulesfloat16hasattrlistpreprocess_modelrr-rload_state_dict"_keys_to_ignore_on_load_unexpectedresearchlenr`postprocess_modelrtoevalr+)3reruromapping_class_namerxdeprecation_messageryrzr{r|r}r~rrrrrrrrrrrrmapping_functionsr/r4config_mapping_kwargsdiffusers_model_config$default_pretrained_model_config_nameexpected_kwargsoptional_kwargsr6 legacy_keynew_keyk model_kwargsctxrcheckpoint_mapping_kwargsdiffusers_format_checkpointrrr param_deviceempty_state_dict param_namer_patdevice_map_kwargss3 rLfrom_single_filez'FromOriginalModelMixin.from_single_files%DESI  %KDIIVrVwVwVyLzK{| )/ 3RTX(Y% ( 4g  5w@S T4Q 1Hd+ **%6=  /"=M  $4e<**Y- 7D)JJ{D1 !::&8$?JJ{D1 ::j$/ **%6=jj5 $jj)>E"JJ':J45-#!1!)% J  *1==>QRL  - - /&99+FK L89KL 12I J  &"&77$56I$J!$(! ( BCUBVW@@/3/"7Zj"k$@AR$]V\$] !%6& /J&J_& "!fc*;A8$G0 ;7=>]7^4&*;; 12E FI%* &)__.R#!1( &5& "03/F/Fs/K ,O_"33 1/ B +8+>+>+@A'J!V+*0**Z*@wA7=m_@TXY]lXlAvzz!},mLm " ) ), 7$5 ; U <OO$:;E <%AAV$aZ`$a! 253C3C3Ez R*?+-*+Ha+ '+5 '**!"4!55yz %($=$=T$I$ EMM ) _glD^._ ! $#&#<#< 2D9(<'=$$&  #  ) )6%9 *  395<</u||E?RL$//1 -H)J^nLn Ol+J %+!%)%9 /   !&!6!67R[`!6!a A  3 3 ??? \.="[3PQARAZ1"["[ \  ! # NNWX[XdXdWeejlpluluwFmGlHkIJ   #  * *5 1!-E   "|'; HH[ !  !!-z :  5 6$5 6 gn < rs " 6"/QQ2>   H %=FD'"'>'@!%!&\ !U\ !<B# }  \ !;A"\ "!>H#\ *!P*+\ 2!L3\ 8!L9\ >!Q*?\ F!S*!G\ N!!P*#O\ V!H"W\ ^-/]^_\ `!R* a\ h%!O*'i\ p!!U*#q\ x !=*"y\ @!F*A\ J"F* "F*" ">" "J*' "T*! "-*$q\ ~XaarN