Spaces:
Running
on
Zero
Running
on
Zero
Update model.py
Browse files
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():
|