Spaces:
Running
Running
Commit
·
07296dc
1
Parent(s):
ebfe92b
Update allow_patterns
Browse files- 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",
|
| 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
|