o }oi@sUddlmZddlmZdZeed<dZeed<dZdZ eed <d Z eed <d Z d eeefdefddZ deeefdefddZ deeefdefddZdS))Path)Unionweights WEIGHTS_PATHcontext CONTEXT_PATHzadapter_metadata.json hf_weightsHF_WEIGHTS_PATH hf_adapterHF_ADAPTER_PATHzadapter_config.jsonbase_dirreturncCs~ddlm}ddlm}m}||r|}||}nt|}|jd|kr*||}t||r=|jjd|kr=|j||_|S)aAppends a given suffix to a base directory path only if it is not already present. This function takes a base directory (either a string or Path) and ensures that the suffix is appended to the path. If the base directory is an AdapterPath instance, it also appends the suffix to the AdapterPath's base_model_path if the suffix is not already part of that path. Args: base_dir (Union[str, Path]): The base directory or path object. suffix (str): The suffix to append to the base directory. Returns: Path: The updated path object with the suffix appended if it was not already present. r AdapterPathimport_multistorageclientis_multistorageclient_url) nemo.lightning.resumernemo.utils.msc_utilsrrrparts isinstancebase_model_path)r suffixrrrmscrM/home/ubuntu/.local/lib/python3.10/site-packages/nemo/lightning/ckpt_utils.pyidempotent_path_appends   rfilepathcCst|d}t|tS)zjGiven an input checkpoint filepath, clean it using `ckpt_to_dir` and then return the context subdirectory.)r) ckpt_to_dirrr)rr rrrckpt_to_context_subdir=s  r cCsddlm}ddlm}m}t||r|S||r"|}||}nt|}|jdks3||jd}|jdks@Jd|d| |j }|S)zPTL considers checkpoints as .ckpt files. This method removes the extension and returns a path to be used as a directory for distributed checkpoints rrrz.ckptz filepath: z must have .ckpt extension) rrrrrrrr with_suffix with_namestem)rrrrrcheckpoint_dirrrrrCs     rN)pathlibrtypingrrstr__annotations__rADAPTER_META_FILENAMEr r HF_ADAPTER_CONFIG_FILENAMErr rrrrrs