guetLzy commited on
Commit
122fcbc
·
verified ·
1 Parent(s): dea7bf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=True)
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(