Spaces:
Running
Running
Commit
·
ebfe92b
1
Parent(s):
6f3b33b
Update
Browse files- 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(
|
| 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
|