haotongl commited on
Commit
7fe9ab6
1 Parent(s): ce8dcd4

Update promptda/promptda.py

Browse files
Files changed (1) hide show
  1. 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="promptda_vitl.ckpt",
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