update codes

config.json

@@ -23,5 +23,9 @@
   "torch_dtype": "float32",
   "transformers_version": "4.47.1",
   "use_cache": true,
-  "vocab_size": 160025
+  "vocab_size": 160025,
+  "auto_map": {
+    "AutoConfig": "configuration_mitre.MitreConfig",
+    "AutoModel": "modeling_mitre.MitreForConditionalGeneration"
+  }
 }

configuration_mitre.py

@@ -58,4 +58,6 @@ class MitreConfig(PretrainedConfig):
             is_encoder_decoder=is_encoder_decoder,
             decoder_start_token_id=decoder_start_token_id,
             **kwargs,
-        )
+        )
+
+MitreConfig.register_for_auto_class("AutoConfig")

modeling_mitre.py

@@ -838,3 +838,4 @@ class MitreForConditionalGeneration(MitrePreTrainedModel, GenerationMixin):
         return sequence_outputs["sequences"]
 
 
+MitreForConditionalGeneration.register_for_auto_class("AutoModel")

tokenizer_config.json

@@ -9,7 +9,6 @@
     "model_max_length": 256,
     "name_or_path": "naist-nlp/mitre_466m",
     "tokenizer_class": "MitreTokenizer",
-    "use_fast": false,
     "auto_map": {
         "AutoTokenizer": ["tokenization_mitre.MitreTokenizer"]
     }