i;ddlmZddlZddlmZddlmZmZddlm Z Gdde Z d d dZ d dd Z d dd Z dd Zy)) annotationsN)Path)hf_hub_downloadsnapshot_download)tqdmc.eZdZdZfdZdfd ZxZS) disabled_tqdmz Class to override `disable` argument in case progress bars are globally disabled. Taken from https://github.com/tqdm/tqdm/issues/619#issuecomment-619639324. c.d|d<t||i|y)NTdisable)super__init__)selfargskwargs __class__s ^/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/sentence_transformers/util/file_io.pyr zdisabled_tqdm.__init__s y $)&)cP t||y#t$r |dk7rYywxYw)zBFix for https://github.com/huggingface/huggingface_hub/issues/1603_lockN)r __delattr__AttributeError)rattrrs rrzdisabled_tqdm.__delattr__s3  G  % w s %%)rstrreturnNone)__name__ __module__ __qualname____doc__r r __classcell__)rs@rr r s *rr c 6tt|d||||S)a} Checks if the given model name or path corresponds to a SentenceTransformer model. Args: model_name_or_path (str): The name or path of the model. token (Optional[Union[bool, str]]): The token to be used for authentication. Defaults to None. cache_folder (Optional[str]): The folder to cache the model files. Defaults to None. revision (Optional[str]): The revision of the model. Defaults to None. local_files_only (bool): Whether to only use local files for the model. Defaults to False. Returns: bool: True if the model is a SentenceTransformer model, False otherwise. z modules.json)token cache_folderrevisionlocal_files_only)boolload_file_path)model_name_or_pathr"r#r$r%s ris_sentence_transformer_modelr)s+(   %-    rc t|||}|jr t|St||} t||j|j j |d|||S#t$rYywxYw)a Loads a file from a local or remote location. Args: model_name_or_path (str): The model name or path. filename (str): The name of the file to load. subfolder (str): The subfolder within the model subfolder (if applicable). token (Optional[Union[bool, str]]): The token to access the remote file (if applicable). cache_folder (Optional[str]): The folder to cache the downloaded file (if applicable). revision (Optional[str], optional): The revision of the file (if applicable). Defaults to None. local_files_only (bool, optional): Whether to only consider local files. Defaults to False. Returns: Optional[str]: The path to the loaded file, or None if the file could not be found or loaded. sentence-transformers)filename subfolderr$ library_namer" cache_dirr%N)rexistsrrnameparentas_posix Exception)r(r,r-r"r#r$r% file_paths rr'r'>s2'H=I9~Y)I  ^^&&//10"-   s4A++ A76A7c .t|tr|j}t||}|jr t |S|||dvr|dndd|||t d} t di|}t||S#t$rd|d<t di|}Y'wxYw) af Loads the subfolder path for a given model name or path. Args: model_name_or_path (str): The name or path of the model. subfolder (str): The subfolder to load. token (Optional[Union[bool, str]]): The token for authentication. cache_folder (Optional[str]): The folder to cache the downloaded files. revision (Optional[str], optional): The revision of the model. Defaults to None. local_files_only (bool, optional): Whether to only use local files. Defaults to False. Returns: Optional[str]: The subfolder path if it exists, otherwise None. ).z/**Nr+)repo_idr$allow_patternsr.r"r/r% tqdm_classTr%) isinstancerr3r0rr rr4) r(r-r"r#r$r%dir_pathdownload_kwargs repo_paths r load_dir_pathrAls,)T"&&( & 2H8}&/8 /IYKs+t/!,# O9%88  9 %% 9.2*+%88 9s! A88BBc  ddl}tjj |dk7r4tj tjj |d|dz}|j d|d 5}|j|jjd }| t|nd}td |dd d tjj|} t|d5}|jdD]0} | s|jt!| |j#| 2 dddtj$|| |j- dddy#t$r tdwxYw#1swYRxYw#t&$r6tjj)|rtj*|wxYw#|j-wxYw#1swYyxYw)aDownload a URL to a local file with a progress bar. The content is streamed in chunks and first written to a temporary ``"_part"`` file, which is atomically moved to ``path`` once the download has completed successfully. Parent directories of ``path`` are created automatically if they do not exist. Args: url (str): The HTTP(S) URL to download. path (str): Destination file path on the local filesystem. Raises: ImportError: If the optional ``httpx`` dependency is not installed. httpx.HTTPStatusError: If the HTTP request returns a non-success status code. OSError: If the file cannot be written to ``path``. Returns: None rNzRhttpx is required to use this function. Please install it via `pip install httpx`.r7T)exist_ok_partGET)follow_redirectszContent-LengthBFz Downloading )unittotal unit_scaleleavedescwbi) chunk_size)httpx ImportErrorospathdirnamemakedirsstreamraise_for_statusheadersgetintrbasenameopen iter_bytesupdatelenwritereplacer4r0removeclose) urlrRrOdownload_filepathresponsecontent_lengthrIprogress file_binarychunks rhttp_getrjs(p wwt" BGGOOD)D9w eS4 8H!!#!))--.>?'5'AN#tEd% UWU\U\UeUefjUkTlFm  '. 1+%00D0A1E E 3#))%01 1 JJ($ / NN ) pnoop  1 1  ww~~/0 +,   NN )sZE-5A+G(! F-F-F4FG(-FF F?GGG%%G((G1)NNNF) r(rr"bool | str | Noner# str | Noner$rlr%r&rr&)r7NNNF)r(rr,z str | Pathr-rr"rkr#rlr$rlr%r&rrl)r(rr-rr"rkr#rlr$rlr%r&rrl)rcrrRrrr) __future__rrQpathlibrhuggingface_hubrrtqdm.autonotebookrr r)r'rArjr<rrrqs*" >"D, $#"       F##"++++  +  +  +++b $#" /&/&/& /& /&  /&  /&/&d1r