Add checkpoints dir load

Nested/utils/helpers.py

@@ -9,7 +9,7 @@ import json
 import random
 import numpy as np
 from argparse import Namespace
-from huggingface_hub import hf_hub_download
+from huggingface_hub import hf_hub_download, snapshot_download
 
 
 def logging_config(log_file=None):
@@ -97,7 +97,9 @@ def load_checkpoint(model_path):
     train_config.trainer_config["kwargs"]["loss"] = loss
 
     tagger = load_object(train_config.trainer_config["fn"], train_config.trainer_config["kwargs"])
-    checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", local_dir="checkpoints")
+    # checkpoint_path = hf_hub_download(repo_id="SinaLab/Nested", local_dir="checkpoints")
+    checkpoint_path = snapshot_download("SinaLab/Nested/checkpoints")
+
 
     tagger.load(checkpoint_path)
     return tagger, tag_vocab, train_config