1;jidxddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlmZddlmZmZmZmZmZmZmZddlmZmZmZddlmZddlmZm Z m!Z!m"Z"m#Z#dd l$m%Z%dd l&m'Z'd d l(m)Z)d dl*m+Z+m,Z,m-Z-m.Z.d dl/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7d dl8m9Z9e9e:Z;ee<j=dz Z>ej?Z@dMde e eAdfdeAfdZB dNdeAde eAdeCdeCde eAdeAde eAde eAde e eDde eCdefd ZEdMd!ed"e eAe eAfdefd#ZFdMd$e eAd%e eAfd&ZGejHIejJd'ejHKejJd(d)d*ZLejHKeLd+ZMdOd,e eAd-e eAddfd.ZNejHKed/ZOejHPeOsdZQnOeReO5ZS eTeSUZQn #eV$rdZQYnwxYwdddn #1swxYwYeQd krejHWeMoeXejYeMdkZZeZre;[d0 eNnc#e\$r[Z]d1Kej^e]j_Z`e;ad2e`d1e]jbj:d3e]d4YdZ][]ndZ][]wwxYweQd krq ejced56eReOd75ZSeSdd8dddn #1swxYwYn$#e\$re;[d9ed:YnwxYwdMd;eAd d?e eAefd;eAd@e eAdAe eAdBeCdCe e dDe eCdEeCde eAde e e eAfdFe eAd%e eAfdGZfdHZg dPdJZhGdKdLZidS)QN)Path)DictListOptionalUnion)uuid4) ModelCard ModelCardData create_repohf_hub_download model_infosnapshot_download upload_folder) HF_HUB_CACHEHF_HUB_DISABLE_TELEMETRYHF_HUB_OFFLINE)REGEX_COMMIT_HASH)EntryNotFoundErrorRepositoryNotFoundErrorRevisionNotFoundErroris_jinja_availablevalidate_hf_hub_args)version) HTTPError) __version__)DEPRECATED_REVISION_ARGSHUGGINGFACE_CO_RESOLVE_ENDPOINTSAFETENSORS_WEIGHTS_NAME WEIGHTS_NAME)ENV_VARS_TRUE_VALUES _flax_version _jax_version_onnxruntime_version_torch_versionis_flax_availableis_onnx_availableis_torch_available) get_loggerzmodel_card_template.md user_agentreturncdtdtjddt}t st r|dzStr |dtz }tr|dtz }|dtz }tr |d tz }tjd d t$vr|d z }t'|t(r8|d d d|Dzz }nt'|t.r|d |zz }|S)zF Formats a user-agent string with basic info about a request. z diffusers/z ; python/rz ; session_id/z; telemetry/offz; torch/z; jax/z; flax/z; onnxruntime/DIFFUSERS_IS_CIz ; is_ci/truez; c3*K|]\}}|d|VdS)/N).0kvs S/root/voice-cloning/.venv/lib/python3.11/site-packages/diffusers/utils/hub_utils.py z"http_user_agent..]s0IIda!zzazzIIIIII)rsysrsplit SESSION_IDrrr)r&r'r$r#r(r%osenvirongetupperr" isinstancedictjoinitemsstr)r+uas r6http_user_agentrFKsk ^k ] ]CK,=,=,?,?,B ] ]Q[ ] ]B&>&%%%* ))))( %|%%% ' '''6 53555 z~~',,22448LLL n*d##  dTYYIIj6F6F6H6HIIIIIII J $ $  dZ Ir8Frepo_id_or_pathtoken is_pipeline from_trainingmodel_description base_modelpromptlicensewidget inferencec ftstd tj||} n{#tt f$rg|r0tjt|d| |||t|} n2t} |rdnd} |d | d }tj| | } YnwxYw| S) a Loads or creates a model card. Args: repo_id_or_path (`str`): The repo id (e.g., "runwayml/stable-diffusion-v1-5") or local path where to look for the model card. token (`str`, *optional*): Authentication token. Will default to the stored token. See https://huggingface.co/settings/token for more details. is_pipeline (`bool`): Boolean to indicate if we're adding tag to a [`DiffusionPipeline`]. from_training: (`bool`): Boolean flag to denote if the model card is being created from a training script. model_description (`str`, *optional*): Model description to add to the model card. Helpful when using `load_or_create_model_card` from a training script. base_model (`str`): Base model identifier (e.g., "stabilityai/stable-diffusion-xl-base-1.0"). Useful for DreamBooth-like training. prompt (`str`, *optional*): Prompt used for training. Useful for DreamBooth-like training. license: (`str`, *optional*): License of the output artifact. Helpful when using `load_or_create_model_card` from a training script. widget (`List[dict]`, *optional*): Widget to accompany a gallery template. inference: (`bool`, optional): Whether to turn on inference widget. Helpful when using `load_or_create_model_card` from a training script. zModelcard rendering is based on Jinja templates. Please make sure to have `jinja` installed before using `load_or_create_model_card`. To install it, please run `pip install Jinja2`.rH diffusers)rN library_namerPrLinstance_promptrO) card_data template_pathrKpipelinemodelNu+This is the model card of a 🧨 diffusers zS that has been pushed on the Hub. This model card has been automatically generated.)rK) r ValueErrorr loadrr from_templater MODEL_CARD_TEMPLATE_PATH) rGrHrIrJrKrLrMrNrOrP model_cardrV components r6load_or_create_model_cardr`csF     ?   a^O5AAA  7 8aaa  a"0'#!,')$*! 7"3   JJ&I&1> wI (%qR[%q%q%q!"0N_```J)a, s6A5B.-B.r^tagsc|jj d|j_|Tt|tr|g}|jj g|j_|D]!}|jj|"|S)z?Populates the `model_card` with library name and optional tags.NrS)datarTr@rDraappend)r^ratags r6populate_model_cardrfs|#+'2 $  dC  6D ?  '#%JO  - -C O ' ' , , , , r8 resolved_file commit_hashc|||Stt|}tjd|}|dS|d}t j|r|ndS)zP Extracts the commit hash from a resolved filename toward a cache file. Nzsnapshots/([^/]+)/r)rDras_posixresearchgroupsrmatch)rgrhrls r6extract_commit_hashros~ 7]++446677M Y,m < > H;;DHr8HF_HOMEXDG_CACHE_HOMEz~/.cache huggingfacerS old_cache_dir new_cache_dircB|t}|t}t|}t|}|dD]}|r|s|||z }|j ddtj || tj ||#t$rtdYwxYwdS)Nz **/blobs/*T)parentsexist_okzCould not create symlink between old cache and new cache. If you use an older version of diffusers again, files will be re-downloaded.)rold_diffusers_cacher expanduserglobis_file is_symlink relative_toparentmkdirr<replacesymlinkOSErrorloggerwarning)rsrt old_blob_path new_blob_paths r6 move_cachers:$ + ''2244M''2244M&++L99    " " =+C+C+E+E )M,E,Em,T,TTM  & &td & C C C J}m 4 4 4  =-8888   ]   sC44$DDzversion_diffusers_cache.txtzThe cache for model files in Diffusers v0.14.0 has moved to a new location. Moving your existing cached models. This is a one-time operation, you can interrupt it or run it later by calling `diffusers.utils.hub_utils.move_cache()`. z5There was a problem when trying to move your cache: z: z Please file an issue at https://github.com/huggingface/diffusers/issues/new/choose, copy paste this whole message and we will do our best to help.T)rww1z?There was a problem when trying to write in your cache folder (z=). Please, ensure the directory exists and can be written to. weights_namevariantc|C|d}|dd|gz|ddz}d|}|S)N.)r:rB)rrsplitss r6 _add_variantrsS##C((y(6"##;6xx'' r8) subfolder cache_dirforce_downloadproxiesresume_downloadlocal_files_onlyrHr+revisionrhpretrained_model_name_or_pathrrrrrrrc Lt|}tj|r|Stj|rtjtj||r"tj||} | S|atjtj|||r#tj|||} | St d|d|d| tvr|tks |tkrtj tj tj tj dkr t|t|| ||||||| || p|  } t!jd| d|d| d | d | d | d t$| S#t!jd | d|d| d| d|dt|| d|d|dt|| dt$YnxYw t||||||||| || p|  } | S#t&$rt |dt($rt | d|dt*$rt |d|dt,$r} t d|d| d} ~ wt.$rt dt0d|d|dt $rt d|d |d!|wxYw)"NzError no file named z found in directory rz0.22.0) filenamerrrrrrHr+rrzLoading the variant z from z via `revision='zG'` is deprecated. Loading instead from `revision='main'` with `variant=z)`. Loading model variants via `revision='z9'` will be removed in diffusers v1. Please use `variant='z '` instead.zYou are loading the variant z^'`. This behavior is deprecated and will be removed in diffusers v1. One should use `variant='z%'` instead. However, it appears that z currently does not have a z file in the 'main' branch of z. The Diffusers team and community would be very grateful if you could open an issue: https://github.com/huggingface/diffusers/issues/new with the title 'z is missing z0' so that the correct variant file can be added.z is not a local folder and is not a valid model identifier listed on 'https://huggingface.co/models' If this is a private repository, make sure to pass a token having permission to this repo with `token` or log in with `huggingface-cli login`.z is not a valid git identifier (branch name, tag name or commit id) that exists for this model name. Check the model page at 'https://huggingface.co/z' for available revisions.& does not appear to have a file named z:There was a specific connection error when trying to load z: We couldn't connect to 'zM' to load this model, couldn't find it in the cached files and it looks like z8 is not the path to a directory containing a file named z or Checkout your internet connection or see how to run the library in offline mode at 'https://huggingface.co/docs/diffusers/installation#offline-mode'.zCan't load the model for 'z'. If you were trying to load it from 'https://huggingface.co/models', make sure you don't have a local directory with the same name. Otherwise, make sure 'z=' is the correct path to a directory containing a file named )rDr<pathisfileisdirrBEnvironmentErrorrr!r rparser base_versionr rwarningswarn FutureWarningrrrrrZr)rrrrrrrrrHr+rrh model_fileerrs r6_get_model_filers< %((E$F$F! w~~344c,, 4 5 5a 7>>"',,'DlSS T T &C\RRJ   "rw~~ GLL6 < P P( ( "&CYP\]]J "i|iiIfiii  0 0 0--AY1Y1Y gmK88EFF'-X`JaJaaa ,1),AA'#1#$3%5)'%4     G8GG;XGGjrGG|DGGowGGrzGGG!"!  E 8E E C`E E rzE E [cE E JgE E DPQ]_gDhDhE E HeE E D a E E o { | H J R o S o S E E E !4 (-%#- /!1%#!0[   J &   "0  %   "ee+Heee  "   "0ggXdggg    "tMjttortt    "f+Jff:Wff6Bfff     ":-J::)F::,8::  s(AG((AH>I A!L#KA L#ctj||fd|D}|D]6}tj|st d|d7dS)NcPg|]"}tj|#Sr2)r<rrB)r3f shards_paths r6 z2_check_if_shards_exist_locally..s)VVVrw||K33VVVr8r5 which is required according to the checkpoint index.)r<rrBexistsrZ) local_dirroriginal_shard_filenamesshard_filenames shard_filers @r6_check_if_shards_exist_locallyrs',,y)44KVVVV=UVVVO% w~~j)) >>j>>>  r8r/c "tj|std|d|dt |d5} t j| } dddn #1swxYwYtt| d } | d} t| d | d<| d | d<tj|| }tj|rt!|| | || fS| }d d g}|sKt#|}| D]9t%fd |jD}|st)|d d: t+|||||||||| }n.#t,$r!}t)dt.d|d|d}~wwxYw|rt!|| | || fS)a For a given model: - download and cache all the shards of a sharded checkpoint if `pretrained_model_name_or_path` is a model ID on the Hub - returns the list of paths to all the shards, as well as some metadata. For the description of each arg, see [`PreTrainedModel.from_pretrained`]. `index_filename` is the full path to the index (downloaded and cached if `pretrained_model_name_or_path` is a model ID on the Hub). zCan't find a checkpoint index (z) in rrN weight_mapmetadataall_checkpoint_keys)rrz*.jsonz*.mdc3*K|] }|jvVdSN) rfilename)r3r4rs r6r7z._get_checkpoint_shard_files..s*$b$b1Z1;%>$b$b$b$b$b$br8rr) rrrrrHrallow_patternsignore_patternsr+rz ' to load z?. You should try again after checking your internet connection.)rrr)r<rrrZopenjsonloadsreadsortedsetvalueslistkeyscopyrBrrr anysiblingsrrrr)rindex_filenamerrrrrHr+rrrindexrsharded_metadatarrrmodel_files_infoshard_file_present cached_folderers @r6_get_checkpoint_shard_filesrs, 7>>. ) )rp>ppPmpppqqq nc " "%a 16688$$%%%%%%%%%%%%%%% &c% *=*D*D*F*F&G&GHHZ(.253F3K3K3M3M.N.N*+%*<%8%=%=%?%?\"',,>>.N(O  %&CDD2  J!$$b$b$b$bHXHa$b$b$b!b!b % &"BB*BBB  ) )+-)+!      >'F > >Ro > > >    &9Og     * **s*'A::A>A>1G G6G11G6ceZdZdZ ddeeejfdedeedeede f d Z ddedeed ee deede d e d eedefdZ dS)PushToHubMixinzR A Mixin to push a model, scheduler, or pipeline to the Hugging Face Hub. NF working_dirrepo_idrHcommit_message create_prc|1d|jjvrd}n d|jjvrd}nd|jj}td|d|d t ||||| S) zB Uploads all files in `working_dir` to `repo_id`. NModelz Upload model SchedulerzUpload schedulerzUpload zUploading the files of z to r)r folder_pathrHrr) __class____name__rinfor)selfrrrHrrs r6_upload_folderzPushToHubMixin._upload_folders  !$.111!/ 777!3!D4>+B!D!D IkIIwIIIJJJER`lu    r8Tprivatesafe_serializationrr,ct|||dj}t||}t|}d|i} d|jjvr| d|itj5} |j | fi| | tj | d|| ||||cd d d S#1swxYwYd S) u Upload model, scheduler, or pipeline files to the 🤗 Hugging Face Hub. Parameters: repo_id (`str`): The name of the repository you want to push your model, scheduler, or pipeline files to. It should contain your organization name when pushing to an organization. `repo_id` can also be a path to a local directory. commit_message (`str`, *optional*): Message to commit while pushing. Default to `"Upload {object}"`. private (`bool`, *optional*): Whether or not the repository created should be private. token (`str`, *optional*): The token to use as HTTP bearer authorization for remote files. The token generated when running `huggingface-cli login` (stored in `~/.huggingface`). create_pr (`bool`, *optional*, defaults to `False`): Whether or not to create a PR with the uploaded files or directly commit. safe_serialization (`bool`, *optional*, defaults to `True`): Whether or not to convert the model weights to the `safetensors` format. variant (`str`, *optional*): If specified, weights are saved in the format `pytorch_model..bin`. Examples: ```python from diffusers import UNet2DConditionModel unet = UNet2DConditionModel.from_pretrained("stabilityai/stable-diffusion-2", subfolder="unet") # Push the `unet` to your namespace with the name "my-finetuned-unet". unet.push_to_hub("my-finetuned-unet") # Push the `unet` to an organization with the name "my-finetuned-unet". unet.push_to_hub("your-org/my-finetuned-unet") ``` T)rrHrwrRrrrz README.md)rHrrN)r rr`rfrrupdatetempfileTemporaryDirectorysave_pretrainedsaver<rrBr) rrrrrHrrrr^ save_kwargstmpdirs r6 push_to_hubzPushToHubMixin.push_to_hubsK\gwedSSS[/weDDD (44 ,-?@ dn5 5 5    73 4 4 4  ( * * f D  7 7; 7 7 7 OOBGLL== > > >&&-# '                  s5ACC!$C!)NNF)NNNFTN) r __module__ __qualname____doc__rrDr<PathLikerboolrrr2r8r6rrs  $(,   3 +,  }  !       6)-"&##'!%EEE! E$ E } E  E!E#E EEEEEEr8rr) NNFFNNNNNN)NN)NNFFNNNr/)jrr<rkr9r tracebackrpathlibrtypingrrrruuidrhuggingface_hubr r r r r rrhuggingface_hub.constantsrrrhuggingface_hub.file_downloadrhuggingface_hub.utilsrrrrr packagingrrequestsrr/r constantsrrr r! import_utilsr"r#r$r%r&r'r(r)loggingr*rr__file__r~r]hexr;rDrFrrAr`rfrorrygetenvrB hf_cache_homerxrcache_version_filer cache_versionrrintrrZrlenlistdirold_cache_is_not_emptyr Exceptionr format_tb __traceback__traceerrorrmakedirswriterrrrrr2r8r6rs1 " ............]\\\\\\\\\;;;;;;                      H  4>>03KK UWW[ dCo 6#2 '+ !#' $CCC C=CC C  } C  C SMCc]C T$Z C~CCCCCL  I U3S >5J V_     I Ix} I8C= I I I I$"" BIiibi0@*&M&M}]]^^ gll=+>>hsm8C=\`,W\\,0MNN w~~())MM  ! !Q CMMMM   MMM  1W]]+>??lCC  SfHgHgDhDhklDl I     JLLLL   II1i1!/BBCCE LL;%;;STS^Sg;;kl;;;         1  L40000 T$c * * a GGCLLL                    :l : : :      sXc]c $# "&*"-1"!%sss#(d#3ss} s } s  sd^sd^ss C=stSy)*ssms#ssssv   T+T+T+T+nbbbbbbbbbbsI!H?>I?I II  III8 KL#ALL#-M9 M-! M9-M11M94M15M99NN