Update modeling.py

modeling.py

@@ -119,10 +119,4 @@ class SmallTransformer(PreTrainedModel):
         return decoder_input_ids
 
 
-# Register the custom model type so AutoModel can find it.
-# This ensures that when AutoModelForSeq2SeqLM sees 'model_type': 'small_transformer' 
-# in your config.json, it knows to use the SmallTransformer class from this file.
-if SmallTransformer.config_class.model_type in PreTrainedModel._model_mapping.keys():
-    pass
-else:
-    PreTrainedModel._model_mapping.register(SmallTransformer.config_class, SmallTransformer)
+# No registration needed - auto_map in config.json handles this