rhJddlmZddlZddlZddlmZddlZddlmZm Z ddl m Z Gdde Z d d dZ d dd Z d dd Zdd 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 u/var/www/html/ai-insurance-compliance-backend/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_pathrAns,)T"&&( & 2H8}&/8 /IYKs+t/!,# O9%88  9 %% 9.2*+%88 9s! A88BBctjj|dk7r4tjtjj|dt j |d}|j dk7rrps0" >"D, $#"       F##"++++  +  +  +++b $#" /&/&/& /& /&  /&  /&/&d"r