Spaces:
Sleeping
Sleeping
Oleg Shulyakov
commited on
Commit
·
1cbbedd
1
Parent(s):
85b89f3
Fix local_dir_use_symlinks warning
Browse files
app.py
CHANGED
|
@@ -161,7 +161,7 @@ def download_base_model(token: str, model_id: str, outdir: tempfile.TemporaryDir
|
|
| 161 |
dl_pattern = ["*.md", "*.json", "*.model"]
|
| 162 |
dl_pattern += [pattern]
|
| 163 |
|
| 164 |
-
api.snapshot_download(repo_id=model_id, local_dir=local_dir,
|
| 165 |
print("Model downloaded successfully!")
|
| 166 |
|
| 167 |
print(f"Model directory contents: {os.listdir(local_dir)}")
|
|
|
|
| 161 |
dl_pattern = ["*.md", "*.json", "*.model"]
|
| 162 |
dl_pattern += [pattern]
|
| 163 |
|
| 164 |
+
api.snapshot_download(repo_id=model_id, local_dir=local_dir, allow_patterns=dl_pattern)
|
| 165 |
print("Model downloaded successfully!")
|
| 166 |
|
| 167 |
print(f"Model directory contents: {os.listdir(local_dir)}")
|