(tia8ddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZmZmZdd lmZmZmZmZmZmZmZmZmZmZmZmZmZej>e Z!d gZ"er ddl#Z#dd l#m$Z$m%Z% dd Z&d Z'dZ( ddZ)GddZ*y)Nsnapshot_download)LocalEntryNotFoundErrorvalidate_hf_hub_args)version)Self) deprecateis_transformers_availablelogging) SingleFileComponentError_is_legacy_scheduler_kwargs"_is_model_weights_in_cached_folder_legacy_load_clip_tokenizer_legacy_load_safety_checker_legacy_load_scheduler$create_diffusers_clip_model_from_ldm)create_diffusers_t5_model_from_checkpointfetch_diffusers_configfetch_original_configis_clip_model_in_single_fileis_t5_in_single_fileload_single_file_checkpointsafety_checker)PreTrainedModelPreTrainedTokenizerc N|rt||} t| |}n!tj|}t||}trAt j t j t jj}nd}txr*t|txr|t j dk\}txr*t|txr|t j dk\}tjtjdd}t||j}t||j}t||j }|r"t|d}|rd}|d||||| || d| }|S|r t#||rt%||||| || }|S|rt'|rt)||||| |}|S|r| rt+|||| }|S|r| s t-| rt/|f|||d | }|St1|d st3d |jd i}|j5|||dt|t6j8j:r|j5d| i|s|rt=||st?d|dt|d }|di|}|S)NN/A4.20.0.rfrom_single_file)%pretrained_model_link_or_path_or_dictoriginal_configconfig subfolder torch_dtypelocal_files_only disable_mmap) checkpointr%r&r'r(is_legacy_loading)r*r%r&r'r()r*r%r()r*component_namer$from_pretrainedzThe component zI cannot be loaded as it does not seem to have a supported loading method.)pretrained_model_name_or_pathr&r(r'zFailed to load zD. Weights for this component appear to be missing in the checkpoint.) getattr importlib import_moduler rparse transformers __version__ base_version issubclassrr__name__splitFromOriginalModelMixin ModelMixinSchedulerMixinrrrrrrrhasattr ValueErrorupdatetorchnnModulerr) library_name class_namenamer* pipelinesis_pipeline_modulecached_model_config_pathr$r(r'r+r)kwargspipeline_module class_objlibrarytransformers_versionis_transformers_model is_tokenizerdiffusers_moduleis_diffusers_single_file_modelis_diffusers_modelis_diffusers_scheduler load_methodloaded_sub_modelloading_kwargss l/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/diffusers/loaders/single_file.pyload_single_file_sub_modelrX4s !)\:OZ8 )),7GZ0  "&}}W]]<;S;S-T-a-ab$ "# < y/ 2 < GMM($; ; "# < y"5 6 < GMM($; ; !..x~~c/B1/EF%/ ;K;b;b%c"#I/?/J/JK' 3C3R3RS%i);<  '+ $&  2<++#-%    T  #? :#V? !+#-/ | i #7 #CD !+#-  f U +6 *5M`p R K %6:UV\:]1  ",TSb fl H ?y"34$Y%7%7$8933 1I!$4   i 1  ! !=+"> ? !656NPTU.%j\1uvi):; &88 ctjtjdd}i}|j ddt rAt jt jtjj}nd}|jD]+\}}t|d|j}|djdk(}t|d|j}t xr-t|dtxr|t jdk\} t xr-t|dt xr|t jdk\} |r|t"vrd|djg||<|s|r#|rddg||<|sd|djg||<| s| r|t"vrd |djg||<%ddg||<.|S) Nr!rselfrKarrasDiffusionSchedulersr diffusers DDIMSchedulerr4)r1r2r8r9popr rr3r4r5r6itemsr7r;r<rrSINGLE_FILE_OPTIONAL_COMPONENTS) component_typesrP config_dictrMr,component_valuerRis_scheduler_enum is_schedulerrNis_transformers_tokenizers rW#_map_component_types_to_config_dictrhs ..x~~c/B1/EFK% "&}}W]]<;S;S-T-a-ab$+:+@+@+B''(:$CC+9?1;M;V;V*WK ',0,K 'E,CH rYcxtj|jj}|j Dcic]&\}}|j tj k(s$||(}}}|j}|j Dcic] \}}||vs ||}}}t|}|Scc}}wcc}}w)N) inspect signature__init__ parametersr`default_empty_get_signature_typesrh)pipeline_classrmkvrequired_parametersrbrcs rW_infer_pipeline_config_dictrus"">#:#:;FFJ,6,<,<,>^,>DAq!))w~~B]1a4,>^$99;O)8(=(=(?\(?11H[C[q!t(?O\5oFK _]s%B0#B0 B6B6c 4gd}t||||||||}|S)N)z **/*.jsonz*.jsonz*.txtz**/*.txtz **/*.model) cache_dirrevisionproxiesforce_downloadr(tokenallow_patternsr) r.rwrxryrzr(r{r|cached_model_paths rW)_download_diffusers_model_config_from_hubr~s5PN)%%)%  rYc0eZdZdZeedefdZy)FromSingleFileMixinzV Load model weights saved in the `.ckpt` format into a [`DiffusionPipeline`]. returnc /|jdd}|jdd}|jdd}|d}tdd||}|jdd}|jd d}|jd d} |jd d} |jd d} |jd d} |jdd} |jdd}d}| Ct| tjs)tj } t jd| d|jdd}|d}tdd|| t|| }ddl m }||d}t|||| | | | |}|t|}|d}n|}tjj!|sH|j#ddkDrt%d|d t'|| | ||| | }|j)|}n|}|j)|}|jd#d|j1|\}}|Dcic]}||vs||j|c}/|Dcic]}||vs||j|}}|j2|fi|\}}}|Dcic]}||vs||j|}}i||}d$d%lm}/fd&} |j9D!cic]\}}!| ||!s||!}}}!t;j<t?|j9d'(D]>\}"\}#}$d}%tA||#}&|"/vr/|"}%n tCd6|#|$|"||&||| || ||d) |}%|%||"<@tI|tI|jKz }(tM/jK})|jN}*tQ|(d$kDr.|(tI|)|*zkr|(D]}+/j|+d||+<nbtQ|(d$kDrTtItM|jKtM/jKz|z })t%d0|d1|d2|)d3|jd4d},|,,d5}td4d|tS| | }-|jU|-|d6i|}.|.S#t*$rr|9t jdt'|| | ||d| }|j)|}n3t jd d!}d}t-|}|j.|d"<YwxYwcc}wcc}wcc}wcc}!}w#tD$r6}'tE|'jFd*|"d+|$d,|j.d-|"d.|"d/ d}'~'wwxYw)7a Instantiate a [`DiffusionPipeline`] from pretrained pipeline weights saved in the `.ckpt` or `.safetensors` format. The pipeline is set in evaluation mode (`model.eval()`) by default. Parameters: pretrained_model_link_or_path (`str` or `os.PathLike`, *optional*): Can be either: - A link to the `.ckpt` file (for example `"https://huggingface.co//blob/main/.ckpt"`) on the Hub. - A path to a *file* containing all pipeline weights. torch_dtype (`str` or `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 (`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. original_config_file (`str`, *optional*): The path to the original config file that was used to train the model. If not provided, the config file will be inferred from the checkpoint file. config (`str`, *optional*): Can be either: - 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. 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. kwargs (remaining dictionary of keyword arguments, *optional*): Can be used to overwrite load and saveable variables (the pipeline components of the specific pipeline class). The overwritten components are passed directly to the pipelines `__init__` method. See example below for more information. Examples: ```py >>> from diffusers import StableDiffusionPipeline >>> # Download pipeline from huggingface.co and cache. >>> pipeline = StableDiffusionPipeline.from_single_file( ... "https://huggingface.co/WarriorMama777/OrangeMixs/blob/main/Models/AbyssOrangeMix/AbyssOrangeMix.safetensors" ... ) >>> # Download pipeline from local file >>> # file is downloaded under ./v1-5-pruned-emaonly.ckpt >>> pipeline = StableDiffusionPipeline.from_single_file("./v1-5-pruned-emaonly.ckpt") >>> # Enable float16 and move to GPU >>> pipeline = StableDiffusionPipeline.from_single_file( ... "https://huggingface.co/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt", ... torch_dtype=torch.float16, ... ) >>> pipeline.to("cuda") ``` original_config_fileNr%r$z`original_config_file` argument is deprecated and will be removed in future versions.please use the `original_config` argument instead.z1.0.0rzFryr{rwr(rxr'r)zPassed `torch_dtype` z7 is not a `torch.dtype`. Defaulting to `torch.float32`.scaling_factorzpPassing the `scaling_factor` argument to `from_single_file is deprecated and will be ignored in future versions.)r(r )_get_pipeline_class)r%)rzryr{rwr(rxr)r./r zThe provided config "zP" is neither a valid local path nor a valid repo id. Please check the parameter.)rwrxryrzr(r{z`local_files_only` is True but no local configs were found for this checkpoint. Attempting to download the necessary config files for this pipeline. a-Detected legacy `from_single_file` loading behavior. Attempting to create the pipeline based on inferred components. This may lead to errors if the model components are not correctly inferred. To avoid this warning, please explicitly pass the `config` argument to `from_single_file` with a path to a local diffusers model repo e.g. `from_single_file(, config=) or run `from_single_file` with `local_files_only=False` first to update the local cache directory with the necessary config files. T _class_name _ignore_filesr)rFc8|dy|vr|y|tvryy)NrFT)ra)rEvaluepassed_class_objs rW load_modulez9FromSingleFileMixin.from_single_file..load_modules6Qx'',, =z) z Pipeline z expected z , but only z were passed.load_safety_checkerzPlease pass instances of `StableDiffusionSafetyChecker` and `AutoImageProcessor`using the `safety_checker` and `feature_extractor` arguments in `from_single_file`r/)+r_r isinstancer@dtypefloat32loggerwarninggetrpipelines.pipeline_utilsrrrospathisdircountr>r~ load_configrrur8_get_signature_keysextract_init_dictr]rFr`r tqdmsortedr=rXrmessagesetkeyslist_optional_componentslenrr?)0clspretrained_model_link_or_pathrIrr%r$deprecation_messagerzryr{rwr(rxr'r)r+rrrqr*$default_pretrained_model_config_namerHrcexpected_modulesoptional_kwargsrrpassed_pipe_kwargs init_dict unused_kwargs_ init_kwargsrFrrsrErCrDrUrGemissing_modulespassed_modulesoptional_modulesmodulersafety_checker_componentspipers0 @rWr"z$FromSingleFileMixin.from_single_filesmT &zz*@$GHd+ **%6=  +E  ,g7J K2O$4e<**Y- 7D)JJ{D1 !::&8%@::j$/jj5 zz.%8 !  ":k5;;+O--K NN' }4kl   $4d;  %:  &1D E  &3OVfgOB,S>0 ))-%   >+J7F39:Y3Z 039 0ww}}AB399#>B +F84ff2 I+T8'%##1%5,(-889QR V(L $(445MNK .,:,N,Ns,S)/6FV6F!v+Avzz!},6FV8GW11PV;aA.W&Fn&F&F{&]V\&]# =!4CVOqqI~q)--**O V;;(:; ' '0oo&7M&7da;q!;LQT&7 M07 9??$ %,L1 ,D,< $ !(L!A ''#3D#9 'A(%1#-!#-+=1I"+$/(7)9*;%1(!($6!1K M1 P./#k6F6F6H2II.3356)>>  ! #3~P`?`;a(a)&6&:&:64&H F#*  !A % k&6&6&8!9DAQAVAVAXE  *e  +W6I J(CDTVa(b %   8 9, , S+% I #*NNa0Y<"+!) ''5).#0,#1"<"<=U"VK NN8)-%/3,"=n"MK1?1H1HK .K% I`WWW N80 2 yyk*! $fC |4&&4&=&=%>>cdhciijkojpq! "   s[#R0 T :T  TT TTT&T>T"A7T  T " U!+1UU!N)r8 __module__ __qualname____doc__ classmethodrrr"r/rYrWrr s-d$ddrYr)NFNFF)NNN)+r1rjrr@huggingface_hubrhuggingface_hub.utilsrr packagingrtyping_extensionsrutilsr r r single_file_utilsrrrrrrrrrrrrr get_loggerr8rrar4rrrXrhrur~rr/rYrWrs -O"AA"   H %$4"4A~B.b " 0kkrY