Skip to content

Commit 20774af

Browse files
authored
ENH: Launch the ChatTTS model with kwargs (#2425)
1 parent 9145e02 commit 20774af

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

xinference/model/audio/chattts.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,8 @@ def load(self):
5353
torch._dynamo.config.suppress_errors = True
5454
torch.set_float32_matmul_precision("high")
5555
self._model = ChatTTS.Chat()
56-
self._model.load(source="custom", custom_path=self._model_path, compile=True)
56+
logger.info("Load ChatTTS model with kwargs: %s", self._kwargs)
57+
self._model.load(source="custom", custom_path=self._model_path, **self._kwargs)
5758

5859
def speech(
5960
self,

xinference/model/audio/tests/test_chattts.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,7 @@ def test_chattts(setup):
2525
client = Client(endpoint)
2626

2727
model_uid = client.launch_model(
28-
model_name="ChatTTS",
29-
model_type="audio",
28+
model_name="ChatTTS", model_type="audio", compile=False
3029
)
3130
model = client.get_model(model_uid)
3231
input_string = (

0 commit comments

Comments
 (0)