Upload folder using huggingface_hub

config.json

@@ -2,6 +2,10 @@
   "architectures": [
     "TaoNetForCausalLM"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_taonet.TaoNetConfig",
+    "AutoModelForCausalLM": "modeling_taonet.TaoNetForCausalLM"
+  },
   "bos_token_id": 1,
   "cnn_channels": [
     32,

export_to_hf.py

@@ -53,6 +53,11 @@ def main():
         eos_token_id=special_tokens.get("<EOS>", 2),
         unk_token_id=special_tokens.get("<UNK>", 0),
     )
+    hf_config.architectures = ["TaoNetForCausalLM"]
+    hf_config.auto_map = {
+        "AutoConfig": "configuration_taonet.TaoNetConfig",
+        "AutoModelForCausalLM": "modeling_taonet.TaoNetForCausalLM",
+    }
 
     model = TaoNetForCausalLM(hf_config)
     missing, unexpected = model.model.load_state_dict(model_state, strict=False)