| 223 | |
| 224 | |
| 225 | def download_hf_model(model: str, revision: Optional[str] = None) -> Path: |
| 226 | ignore_patterns = ["original/**/*"] |
| 227 | logger.info(f"Downloading model {model} from HuggingFace") |
| 228 | with get_file_lock(model): |
| 229 | hf_folder = snapshot_download( |
| 230 | model, |
| 231 | local_files_only=huggingface_hub.constants.HF_HUB_OFFLINE, |
| 232 | ignore_patterns=ignore_patterns, |
| 233 | revision=revision, |
| 234 | tqdm_class=DisabledTqdm) |
| 235 | logger.info(f"Finished downloading model {model} from HuggingFace") |
| 236 | return Path(hf_folder) |
| 237 | |
| 238 | |
| 239 | def download_hf_pretrained_config(model: str, |