Update modeling_emotion_classifier.py

modeling_emotion_classifier.py

@@ -9,7 +9,8 @@ class EmotionClassifierHuBERT(PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        self.hubert = HubertModel.from_pretrained("facebook/hubert-large-ls960-ft")
+        #self.hubert = HubertModel.from_pretrained("facebook/hubert-large-ls960-ft")
+        self.hubert = HubertModel(config.hubert_config)
         self.conv1 = nn.Conv1d(in_channels=1024, out_channels=512, kernel_size=3, padding=1)
         self.conv2 = nn.Conv1d(in_channels=512, out_channels=256, kernel_size=3, padding=1)
         self.transformer_encoder = nn.TransformerEncoderLayer(d_model=256, nhead=8)