Update model.py

model.py

@@ -13,7 +13,7 @@ class Wav2Vec2BERT_Llama(nn.Module):
         super().__init__()
 
         # 1. 加载预训练模型
-        self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained("facebook/w2v-bert-2.0")
+        self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained("facebook/w2v-bert-2.0", output_hidden_states=True)
         
         # 2. 选择性冻结参数
         for name, param in self.wav2vec2bert.named_parameters():