Update app.py
Browse files
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 和
|
| 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 庫版本。")
|