Upload model

config.json

@@ -11,6 +11,7 @@
   "include_time_delta": true,
   "index_value_encoder_intermediate_size": 2048,
   "model_type": "cxrmate-ed",
+  "pad_token_id": 4,
   "prompt_report_sections_filter": [
     "indication",
     "history"

configuration_cxrmate_ed.py

@@ -19,7 +19,7 @@ class CXRMateEDConfig(transformers.PretrainedConfig):
         pad_token_id: int = 4,
         **kwargs,
     ):
-        
+        super().__init__(**kwargs)
         self.vision_config = vision_config
         self.text_config = text_config
         self.index_value_encoder_intermediate_size = index_value_encoder_intermediate_size
@@ -31,7 +31,6 @@ class CXRMateEDConfig(transformers.PretrainedConfig):
         self.prompt_report_sections_filter = prompt_report_sections_filter
         self.pad_token_id = pad_token_id      
 
-        transformers.PretrainedConfig.__init__(self, **kwargs)
 
 # class CXRMateEDConfig(transformers.PretrainedConfig):
 

generation_config.json

@@ -2,5 +2,6 @@
   "_from_model_config": true,
   "bos_token_id": 1,
   "eos_token_id": 2,
+  "pad_token_id": 4,
   "transformers_version": "4.47.0"
 }