From 7e33889f5222bb2f7f602f3af1b68db9577bf4bc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?= <41315874+fumiama@users.noreply.github.com> Date: Fri, 9 Aug 2024 21:35:28 +0800 Subject: [PATCH] optimize(core): compile=False by default on load() --- ChatTTS/core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ChatTTS/core.py b/ChatTTS/core.py index 994eb7741..79ef9bc8d 100644 --- a/ChatTTS/core.py +++ b/ChatTTS/core.py @@ -121,7 +121,7 @@ def load( self, source: Literal["huggingface", "local", "custom"] = "local", force_redownload=False, - compile: bool = True, + compile: bool = False, custom_path: Optional[torch.serialization.FILE_LIKE] = None, device: Optional[torch.device] = None, coef: Optional[torch.Tensor] = None, @@ -229,7 +229,7 @@ def _load( decoder_ckpt_path: str = None, tokenizer_path: str = None, device: Optional[torch.device] = None, - compile: bool = True, + compile: bool = False, coef: Optional[str] = None, use_flash_attn=False, use_vllm=False,