Training in progress - step 20

asr_modeling.py

@@ -851,8 +851,19 @@ class ASRModel(PreTrainedModel):
         if hasattr(self.encoder.config, "num_mel_bins"):
             self.config.audio_config.num_mel_bins = self.encoder.config.num_mel_bins
 
-        # Use parent class to save config and model.safetensors
-        super().save_pretrained(save_dir, **kwargs)
+        # Temporarily remove non-serializable attributes before calling parent save
+        feature_extractor = self.feature_extractor
+        tokenizer = self.tokenizer
+        del self.feature_extractor
+        del self.tokenizer
+
+        try:
+            # Use parent class to save config and model.safetensors
+            super().save_pretrained(save_dir, **kwargs)
+        finally:
+            # Restore attributes
+            self.feature_extractor = feature_extractor
+            self.tokenizer = tokenizer
 
         self.tokenizer.save_pretrained(save_dir)