fix: set missing comple=False (#713)

This commit is contained in:
源文雨
2024-08-24 00:53:33 +08:00
parent 554554c591
commit 7f54d2cd17
2 changed files with 4 additions and 5 deletions
+2 -3
View File
@@ -4,7 +4,6 @@ import os
import sys
import numpy as np
import torch
import ChatTTS
from tools.logger import get_logger
@@ -66,9 +65,9 @@ def main(
is_load = False
if os.path.isdir(custom_path) and source == "custom":
is_load = chat.load(compile=True, source="custom", custom_path=custom_path)
is_load = chat.load(source="custom", custom_path=custom_path)
else:
is_load = chat.load(compile=True, source=source)
is_load = chat.load(source=source)
if is_load:
logger.info("Models loaded successfully.")
+2 -2
View File
@@ -61,11 +61,11 @@ def on_audio_seed_change(audio_seed_input):
def load_chat(cust_path: Optional[str], coef: Optional[str]) -> bool:
if cust_path == None:
ret = chat.load(coef=coef, compile=sys.platform != "win32")
ret = chat.load(coef=coef)
else:
logger.info("local model path: %s", cust_path)
ret = chat.load(
"custom", custom_path=cust_path, coef=coef, compile=sys.platform != "win32"
"custom", custom_path=cust_path, coef=coef
)
global custom_path
custom_path = cust_path