add model

config.json

@@ -3,7 +3,7 @@
     "SegVolModel"
   ],
   "auto_map": {
-    "AutoConfig": "config_segvol.SegVolConfig",
+    "AutoConfig": "model_segvol_single.SegVolConfig",
     "AutoModel": "model_segvol_single.SegVolModel"
   },
   "model_type": "segvol",

config_segvol.py

@@ -1,14 +0,0 @@
-from transformers import PretrainedConfig
-
-class SegVolConfig(PretrainedConfig):
-    model_type = "segvol"
-
-    def __init__(
-        self,
-        test_mode=True,
-        **kwargs,
-    ):
-        self.spatial_size = [32, 256, 256]
-        self.patch_size = [4, 16, 16]
-        self.test_mode = test_mode
-        super().__init__(**kwargs)

model_segvol_single.py

@@ -1,8 +1,21 @@
-from transformers import PreTrainedModel
+from transformers import PreTrainedModel, PretrainedConfig
 from config_segvol import SegVolConfig
 import numpy as np
 import monai.transforms as transforms
 
+class SegVolConfig(PretrainedConfig):
+    model_type = "segvol"
+
+    def __init__(
+        self,
+        test_mode=True,
+        **kwargs,
+    ):
+        self.spatial_size = [32, 256, 256]
+        self.patch_size = [4, 16, 16]
+        self.test_mode = test_mode
+        super().__init__(**kwargs)
+        
 class SegVolModel(PreTrainedModel):
     config_class = SegVolConfig