gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-12 15:50:07 -08:00
Code Issues Packages Projects Releases Wiki Activity

add device option for infer-cli, patch-1

Browse Source
This commit is contained in:
SWivid
2025-03-22 17:35:16 +08:00
parent 1d82b7928e
commit 1d0cf2b8ba
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/f5_tts/infer/infer_cli.py
View File

@@ -22,6 +22,7 @@ from f5_tts.infer.utils_infer import (
sway_sampling_coef,
speed,
fix_duration,
device,
infer_process,
load_model,
load_vocoder,
@@ -207,7 +208,7 @@ cfg_strength = args.cfg_strength or config.get("cfg_strength", cfg_strength)
sway_sampling_coef = args.sway_sampling_coef or config.get("sway_sampling_coef", sway_sampling_coef)
speed = args.speed or config.get("speed", speed)
fix_duration = args.fix_duration or config.get("fix_duration", fix_duration)
device = args.device
device = args.device or config.get("device", device)
# patches for pip pkg user