Spaces:
Running
on
Zero
Running
on
Zero
Update promptda/promptda.py
Browse files- promptda/promptda.py +1 -1
promptda/promptda.py
CHANGED
@@ -62,7 +62,7 @@ class PromptDA(nn.Module):
|
|
62 |
cached_checkpoint_path = hf_hub_download(
|
63 |
repo_id=pretrained_model_name_or_path,
|
64 |
repo_type="model",
|
65 |
-
filename="
|
66 |
**hf_kwargs
|
67 |
)
|
68 |
ckpt_path = cached_checkpoint_path
|
|
|
62 |
cached_checkpoint_path = hf_hub_download(
|
63 |
repo_id=pretrained_model_name_or_path,
|
64 |
repo_type="model",
|
65 |
+
filename="model.ckpt",
|
66 |
**hf_kwargs
|
67 |
)
|
68 |
ckpt_path = cached_checkpoint_path
|