TymaaHammouda commited on
Commit
6f3b33b
·
1 Parent(s): fec514c

Add snapshot_download

Browse files
Files changed (1) hide show
  1. Nested/utils/helpers.py +1 -1
Nested/utils/helpers.py CHANGED
@@ -98,7 +98,7 @@ def load_checkpoint(model_path):
98
 
99
  tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
100
  # checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", local_dir="checkpoints")
101
- checkpoint_path = snapshot_download("SinaLab/Nested/checkpoints")
102
 
103
 
104
  tagger.load(checkpoint_path)
 
98
 
99
  tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
100
  # checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", local_dir="checkpoints")
101
+ checkpoint_path = snapshot_download(model_id="SinaLab/Nested", cache_dir="checkpoints")
102
 
103
 
104
  tagger.load(checkpoint_path)