Skip to content

Commit b3fd560

Browse files
authored
Update f5_tts_trtllm.py
1 parent 5f22753 commit b3fd560

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/f5_tts/runtime/triton_trtllm/model_repo_f5_tts/f5_tts/1/f5_tts_trtllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -236,7 +236,7 @@ def __init__(
236236
self.rope_cos = self.freqs.cos().half()
237237
self.rope_sin = self.freqs.sin().half()
238238

239-
self.nfe_steps = 16
239+
self.nfe_steps = 32
240240
epss = {
241241
5: [0, 2, 4, 8, 16, 32],
242242
6: [0, 2, 4, 6, 8, 16, 32],

0 commit comments

Comments
 (0)