UniqAI/openedai-speech
1
0
Fork 0
mirror of https://github.com/matatonic/openedai-speech synced 2025-06-26 18:16:32 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

0.9.0 + fix xtts not None

Browse Source
This commit is contained in:
matatonic 2024-04-23 22:35:03 -04:00
parent 4d76aca1af
commit a2a3d2b3eb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
speech.py
View File

@ -150,11 +150,12 @@ async def generate_speech(request: GenerateSpeechRequest):
if xtts is not None and xtts.model_name != tts_model:
import torch, gc
del xtts
xtts = None
gc.collect()
torch.cuda.empty_cache()
if 'parler-tts' in tts_model and has_parler_tts:
if not xtts:
if xtts is None:
xtts = parler_tts(tts_model, device=args.xtts_device)
ffmpeg_args = build_ffmpeg_args(response_format, input_format="WAV", sample_rate=str(xtts.model.config.sampling_rate))
@ -165,7 +166,7 @@ async def generate_speech(request: GenerateSpeechRequest):
tts_io_out = xtts.tts(text=input_text, description=speaker)
else:
if not xtts:
if xtts is None:
xtts = xtts_wrapper(tts_model, device=args.xtts_device)
ffmpeg_args = build_ffmpeg_args(response_format, input_format="WAV", sample_rate="24000")