Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -75,7 +75,7 @@ def load_hubert():
|
|
75 |
"""加载HuBERT模型(使用fairseq格式的检查点)"""
|
76 |
print("正在加载 HuBERT 模型...")
|
77 |
fairseq_path = os.path.join("HuBERT", "chinese-hubert-large-fairseq-ckpt.pt")
|
78 |
-
state_dict = torch.load(fairseq_path, map_location="cpu",weights_only=
|
79 |
|
80 |
# 加载标准HuBERT模型
|
81 |
model = HubertModel.from_pretrained(
|
|
|
75 |
"""加载HuBERT模型(使用fairseq格式的检查点)"""
|
76 |
print("正在加载 HuBERT 模型...")
|
77 |
fairseq_path = os.path.join("HuBERT", "chinese-hubert-large-fairseq-ckpt.pt")
|
78 |
+
state_dict = torch.load(fairseq_path, map_location="cpu",weights_only=False)
|
79 |
|
80 |
# 加载标准HuBERT模型
|
81 |
model = HubertModel.from_pretrained(
|