Update wav2vec2speechclassification.py

wav2vec2speechclassification.py

@@ -26,6 +26,7 @@ from transformers.models.wav2vec2.modeling_wav2vec2 import (
 class Wav2Vec2ClassificationHead(nn.Module):
     """Head for wav2vec classification task."""
     config_class = Wav2Vec2Config
+    model_type = "wav2vec2"
 
     def __init__(self, config):
         super().__init__()
@@ -45,6 +46,7 @@ class Wav2Vec2ClassificationHead(nn.Module):
 
 class Wav2Vec2ForSpeechClassification(Wav2Vec2PreTrainedModel):
     config_class = Wav2Vec2Config
+    model_type = "wav2vec2"
     
     def __init__(self, config):
         super().__init__(config)