TymaaHammouda commited on
Commit
07296dc
·
1 Parent(s): ebfe92b

Update allow_patterns

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(repo_id="SinaLab/Nested", repo_type="checkpoints")
102
 
103
  tagger.load(checkpoint_path)
104
  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", allow_patterns="checkpoints/")
102
 
103
  tagger.load(checkpoint_path)
104
  return tagger, tag_vocab, train_config