Ryanus commited on
Commit
a403eae
·
verified ·
1 Parent(s): 92c02f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -14,12 +14,13 @@ os.environ["COQUI_TOS_AGREED"] = "1"
14
  import torch.serialization
15
  from TTS.tts.configs.xtts_config import XttsConfig
16
  from TTS.tts.models.xtts import XttsAudioConfig
17
- from TTS.config.shared_configs import BaseDatasetConfig # 新增這行!
 
18
 
19
  try:
20
  # 將所有可能導致問題的類都加入白名單
21
- torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig]) # 更新這行!
22
- print("已將 XttsConfig, XttsAudioConfig 和 BaseDatasetConfig 加入 PyTorch 安全全局變數白名單。")
23
  except Exception as e:
24
  print(f"警告:無法將安全全局變數加入 PyTorch 白名單: {e}")
25
  print("如果遇到模型載入錯誤,請檢查 PyTorch 和 TTS 庫版本。")
 
14
  import torch.serialization
15
  from TTS.tts.configs.xtts_config import XttsConfig
16
  from TTS.tts.models.xtts import XttsAudioConfig
17
+ from TTS.config.shared_configs import BaseDatasetConfig
18
+ from TTS.tts.models.xtts import XttsArgs # 新增這行!
19
 
20
  try:
21
  # 將所有可能導致問題的類都加入白名單
22
+ torch.serialization.add_safe_globals([XttsConfig, XttsAudioConfig, BaseDatasetConfig, XttsArgs]) # 更新這行!
23
+ print("已將 XttsConfig, XttsAudioConfig, BaseDatasetConfig XttsArgs 加入 PyTorch 安全全局變數白名單。")
24
  except Exception as e:
25
  print(f"警告:無法將安全全局變數加入 PyTorch 白名單: {e}")
26
  print("如果遇到模型載入錯誤,請檢查 PyTorch 和 TTS 庫版本。")