TymaaHammouda commited on
Commit
ebfe92b
·
1 Parent(s): 6f3b33b
Files changed (1) hide show
  1. Nested/utils/helpers.py +1 -2
Nested/utils/helpers.py CHANGED
@@ -98,8 +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(model_id="SinaLab/Nested", cache_dir="checkpoints")
102
-
103
 
104
  tagger.load(checkpoint_path)
105
  return tagger, tag_vocab, train_config
 
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(repo_id="SinaLab/Nested", repo_type="checkpoints")
 
102
 
103
  tagger.load(checkpoint_path)
104
  return tagger, tag_vocab, train_config