Update model.py

model.py

@@ -5,14 +5,22 @@ from llama_nar import LlamaNAREmb
 from transformers import LlamaConfig
 import time
 import torch.nn.functional as F
+from huggingface_hub import hf_hub_download
 
 
 class Wav2Vec2BERT_Llama(nn.Module):
     def __init__(self):
         super().__init__()
 
+        ckpt_path = hf_hub_download(
+            repo_id="amphion/deepfake_detection", 
+            filename="w2vbert2"
+        )
         # 1. 加载预训练模型
-        self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained("/mntcephfs/lab_data/wangli/pretrain/w2v-bert-2.0/", output_hidden_states=True)
+        self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained(
+            ckpt_path, 
+            output_hidden_states=True
+        )
         
         # 2. 选择性冻结参数
         for name, param in self.wav2vec2bert.named_parameters():