Skip to content

Commit 319d037

Browse files
committed
feat(vLLM): add missing params in refine_text
1 parent 4a1962b commit 319d037

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ChatTTS/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -635,7 +635,7 @@ def _refine_text(
635635
input_ids_list = [i.tolist() for i in input_ids]
636636
del input_ids
637637

638-
result = gpt.llm.generate(None, sample_params, input_ids_list)
638+
result = gpt.llm.generate(None, sample_params, input_ids_list, params.show_tqdm)
639639
token_ids = []
640640
hidden_states = []
641641
for i in result:

0 commit comments

Comments
 (0)