| """Find a local folder or a Hugging Face repo that holds Mini's files.""" | |
| from pathlib import Path | |
| def resolve_pretrained_folder(path_or_repo): | |
| """Return a directory with config.json. Download a Hub repo id first.""" | |
| folder = Path(path_or_repo) | |
| if folder.is_dir(): | |
| return folder | |
| from huggingface_hub import snapshot_download | |
| downloaded = snapshot_download(repo_id=str(path_or_repo)) | |
| return Path(downloaded) | |