Upload folder using huggingface_hub

config.json

@@ -2,10 +2,6 @@
   "architectures": [
     "INF5Model"
   ],
-  "auto_map": {
-    "AutoConfig": "model.INF5Config",
-    "AutoModel": "model.INF5Model"
-  },
   "ckpt_path": "checkpoints/model_best.pt",
   "model_type": "inf5",
   "remove_sil": true,

model.py

@@ -1,3 +1,9 @@
+import sys
+import os
+
+current_dir = os.path.dirname(os.path.abspath(__file__))
+sys.path.append(current_dir)
+
 from transformers import PreTrainedModel, PretrainedConfig
 import torch
 import numpy as np