Spaces:
Running
Running
Commit
·
6f3b33b
1
Parent(s):
fec514c
Add snapshot_download
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("SinaLab/Nested
|
| 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)
|