Spaces:
Running on Zero
Running on Zero
fix path exists error
Browse files
utils/download_weights.py
CHANGED
|
@@ -16,7 +16,7 @@ def download_weights():
|
|
| 16 |
checkpoint_dir = Path.home() / ".foundry" / "checkpoints"
|
| 17 |
for model in MODELS:
|
| 18 |
model_path = os.path.join(checkpoint_dir, model+".ckpt")
|
| 19 |
-
if
|
| 20 |
print(f"{model} already exists at {model_path}, skipping download.")
|
| 21 |
else:
|
| 22 |
cmd = f"foundry install {model} --checkpoint-dir {checkpoint_dir}"
|
|
|
|
| 16 |
checkpoint_dir = Path.home() / ".foundry" / "checkpoints"
|
| 17 |
for model in MODELS:
|
| 18 |
model_path = os.path.join(checkpoint_dir, model+".ckpt")
|
| 19 |
+
if os.path.exists(model_path):
|
| 20 |
print(f"{model} already exists at {model_path}, skipping download.")
|
| 21 |
else:
|
| 22 |
cmd = f"foundry install {model} --checkpoint-dir {checkpoint_dir}"
|