refactor: rename PretrainedMixin to AutoModelBase

CodeWithKyrian · CodeWithKyrian · commit 688ef6cf1e80 · 2025-07-18T09:28:58.000+01:00
diff --git a/src/Models/Auto/AutoModel.php b/src/Models/Auto/AutoModel.php
@@ -4,7 +4,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModel extends PretrainedMixin
+class AutoModel extends AutoModelBase
 {
     const ENCODER_ONLY_MODEL_MAPPING = [
         "albert" => \Codewithkyrian\Transformers\Models\Pretrained\AlbertModel::class,
@@ -64,4 +64,4 @@ class AutoModel extends PretrainedMixin
 
 
     const BASE_IF_FAIL = true;
-}
+}
diff --git a/src/Models/Auto/AutoModelBase.php b/src/Models/Auto/AutoModelBase.php
@@ -15,7 +15,7 @@
  * Base class of all AutoModels. Contains the `from_pretrained` function
  * which is used to instantiate pretrained models.
  */
-abstract class PretrainedMixin
+abstract class AutoModelBase
 {
     /**
      * Mapping from model type to model class.
diff --git a/src/Models/Auto/AutoModelForAudioClassification.php b/src/Models/Auto/AutoModelForAudioClassification.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForAudioClassification  extends PretrainedMixin
+class AutoModelForAudioClassification  extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'audio-spectrogram-transformer' => \Codewithkyrian\Transformers\Models\Pretrained\ASTForAudioClassification::class,
@@ -15,4 +15,4 @@ class AutoModelForAudioClassification  extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForCTC.php b/src/Models/Auto/AutoModelForCTC.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForCTC extends PretrainedMixin
+class AutoModelForCTC extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'wav2vec2' => \Codewithkyrian\Transformers\Models\Pretrained\Wav2Vec2ForCTC::class,
@@ -14,4 +14,4 @@ class AutoModelForCTC extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForCausalLM.php b/src/Models/Auto/AutoModelForCausalLM.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForCausalLM extends PretrainedMixin
+class AutoModelForCausalLM extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'gpt2' => \Codewithkyrian\Transformers\Models\Pretrained\GPT2LMHeadModel::class,
@@ -20,4 +20,4 @@ class AutoModelForCausalLM extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForImageClassification.php b/src/Models/Auto/AutoModelForImageClassification.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForImageClassification extends PretrainedMixin
+class AutoModelForImageClassification extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'vit' => \Codewithkyrian\Transformers\Models\Pretrained\ViTForImageClassification::class,
@@ -15,4 +15,4 @@ class AutoModelForImageClassification extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForImageFeatureExtraction.php b/src/Models/Auto/AutoModelForImageFeatureExtraction.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForImageFeatureExtraction extends PretrainedMixin
+class AutoModelForImageFeatureExtraction extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'clip' => \Codewithkyrian\Transformers\Models\Pretrained\CLIPVisionModelWithProjection::class,
@@ -17,4 +17,4 @@ class AutoModelForImageFeatureExtraction extends PretrainedMixin
         AutoModel::ENCODER_ONLY_MODEL_MAPPING,
         AutoModel::DECODER_ONLY_MODEL_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForImageToImage.php b/src/Models/Auto/AutoModelForImageToImage.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForImageToImage extends PretrainedMixin
+class AutoModelForImageToImage extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'swin2sr' => \Codewithkyrian\Transformers\Models\Pretrained\Swin2SRForImageSuperResolution::class,
@@ -14,4 +14,4 @@ class AutoModelForImageToImage extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForMaskedLM.php b/src/Models/Auto/AutoModelForMaskedLM.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForMaskedLM extends PretrainedMixin
+class AutoModelForMaskedLM extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         "albert" => \Codewithkyrian\Transformers\Models\Pretrained\AlbertForMaskedLM::class,
@@ -21,4 +21,4 @@ class AutoModelForMaskedLM extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForObjectDetection.php b/src/Models/Auto/AutoModelForObjectDetection.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForObjectDetection extends PretrainedMixin
+class AutoModelForObjectDetection extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'detr' => \Codewithkyrian\Transformers\Models\Pretrained\DetrForObjectDetection::class,
@@ -15,5 +15,4 @@ class AutoModelForObjectDetection extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-
-}
+}
diff --git a/src/Models/Auto/AutoModelForQuestionAnswering.php b/src/Models/Auto/AutoModelForQuestionAnswering.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForQuestionAnswering extends PretrainedMixin
+class AutoModelForQuestionAnswering extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'albert' => \Codewithkyrian\Transformers\Models\Pretrained\AlbertForQuestionAnswering::class,
@@ -21,4 +21,4 @@ class AutoModelForQuestionAnswering extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForSeq2SeqLM.php b/src/Models/Auto/AutoModelForSeq2SeqLM.php
@@ -4,7 +4,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForSeq2SeqLM extends PretrainedMixin
+class AutoModelForSeq2SeqLM extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'bart' => \Codewithkyrian\Transformers\Models\Pretrained\BartForConditionalGeneration::class,
@@ -15,4 +15,4 @@ class AutoModelForSeq2SeqLM extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForSequenceClassification.php b/src/Models/Auto/AutoModelForSequenceClassification.php
@@ -6,7 +6,7 @@
 namespace Codewithkyrian\Transformers\Models\Auto;
 
 
-class AutoModelForSequenceClassification extends PretrainedMixin
+class AutoModelForSequenceClassification extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'albert' => \Codewithkyrian\Transformers\Models\Pretrained\AlbertForSequenceClassification::class,
@@ -23,4 +23,4 @@ class AutoModelForSequenceClassification extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForSpeechSeq2Seq.php b/src/Models/Auto/AutoModelForSpeechSeq2Seq.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForSpeechSeq2Seq extends PretrainedMixin
+class AutoModelForSpeechSeq2Seq extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         "whisper" => \Codewithkyrian\Transformers\Models\Pretrained\WhisperForConditionalGeneration::class,
@@ -14,4 +14,4 @@ class AutoModelForSpeechSeq2Seq extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForTokenClassification.php b/src/Models/Auto/AutoModelForTokenClassification.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForTokenClassification extends PretrainedMixin
+class AutoModelForTokenClassification extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         "bert" => \Codewithkyrian\Transformers\Models\Pretrained\BertForTokenClassification::class,
@@ -18,4 +18,4 @@ class AutoModelForTokenClassification extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForVision2Seq.php b/src/Models/Auto/AutoModelForVision2Seq.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForVision2Seq extends PretrainedMixin
+class AutoModelForVision2Seq extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'vision-encoder-decoder' => \Codewithkyrian\Transformers\Models\Pretrained\VisionEncoderDecoderModel::class
@@ -14,4 +14,4 @@ class AutoModelForVision2Seq extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-}
+}
diff --git a/src/Models/Auto/AutoModelForZeroShotObjectDetection.php b/src/Models/Auto/AutoModelForZeroShotObjectDetection.php
@@ -5,7 +5,7 @@
 
 namespace Codewithkyrian\Transformers\Models\Auto;
 
-class AutoModelForZeroShotObjectDetection extends PretrainedMixin
+class AutoModelForZeroShotObjectDetection extends AutoModelBase
 {
     const MODEL_CLASS_MAPPING = [
         'owlvit' => \Codewithkyrian\Transformers\Models\Pretrained\OwlViTForObjectDetection::class,
@@ -15,5 +15,4 @@ class AutoModelForZeroShotObjectDetection extends PretrainedMixin
     const MODEL_CLASS_MAPPINGS = [
         self::MODEL_CLASS_MAPPING,
     ];
-
-}
+}

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,7 @@`
`15`	`15`	* Base class of all AutoModels. Contains the `from_pretrained` function
`16`	`16`	`* which is used to instantiate pretrained models.`
`17`	`17`	`*/`
`18`		`-abstract class PretrainedMixin`
	`18`	`+abstract class AutoModelBase`
`19`	`19`	`{`
`20`	`20`	`/**`
`21`	`21`	`* Mapping from model type to model class.`