Spaces:
Running
Running
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)}")
|