wli3221134 commited on
Commit
5d63aaf
·
verified ·
1 Parent(s): 53d1d01

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -8
model.py CHANGED
@@ -12,15 +12,8 @@ class Wav2Vec2BERT_Llama(nn.Module):
12
  def __init__(self):
13
  super().__init__()
14
 
15
- ckpt_path = hf_hub_download(
16
- repo_id="amphion/deepfake_detection",
17
- filename="w2vbert2"
18
- )
19
  # 1. 加载预训练模型
20
- self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained(
21
- ckpt_path,
22
- output_hidden_states=True
23
- )
24
 
25
  # 2. 选择性冻结参数
26
  for name, param in self.wav2vec2bert.named_parameters():
 
12
  def __init__(self):
13
  super().__init__()
14
 
 
 
 
 
15
  # 1. 加载预训练模型
16
+ self.wav2vec2bert = Wav2Vec2BertModel.from_pretrained("facebook/w2v-bert-2.0")
 
 
 
17
 
18
  # 2. 选择性冻结参数
19
  for name, param in self.wav2vec2bert.named_parameters():