Spaces:
Running
Running
Commit
·
7ef344d
1
Parent(s):
1f4d4a1
Migrate debug
Browse files- download_model.py +2 -1
download_model.py
CHANGED
|
@@ -11,7 +11,8 @@ model_path = snapshot_download(repo_id=MODEL_REPO, cache_dir=MODEL_CACHE_DIR)
|
|
| 11 |
print("Model path now: ", model_path)
|
| 12 |
|
| 13 |
# Find the correct snapshot directory
|
| 14 |
-
snapshots_dir = os.path.join(model_path, "snapshots") # snapshot
|
|
|
|
| 15 |
if os.path.exists(snapshots_dir):
|
| 16 |
snapshot_folders = os.listdir(snapshots_dir)
|
| 17 |
if snapshot_folders:
|
|
|
|
| 11 |
print("Model path now: ", model_path)
|
| 12 |
|
| 13 |
# Find the correct snapshot directory
|
| 14 |
+
# snapshots_dir = os.path.join(model_path, "snapshots") # snapshot
|
| 15 |
+
snapshots_dir = model_path
|
| 16 |
if os.path.exists(snapshots_dir):
|
| 17 |
snapshot_folders = os.listdir(snapshots_dir)
|
| 18 |
if snapshot_folders:
|