engineering/uncloseai-speech
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

xtts: +AMD gpu ROCm, +Apple MPS

Browse source
This commit is contained in:
matatonic 2024-06-24 20:35:07 -04:00
parent e9bbc56523
commit 72c7b799b9
12 changed files with 89 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

14
requirements-rocm.txt Normal file
View file

@ -0,0 +1,14 @@
fastapi
uvicorn
loguru
# piper-tts
piper-tts==1.2.0
# xtts
TTS
# XXX, 3.8+ has some issue for now
spacy==3.7.4
# torch==2.2.2 Fixes: https://github.com/matatonic/openedai-speech/issues/9
# Re: https://github.com/pytorch/pytorch/issues/121834
torch==2.2.2; --index-url https://download.pytorch.org/whl/rocm5.7; sys_platform == "linux"
torchaudio==2.2.2; --index-url https://download.pytorch.org/whl/rocm5.7; sys_platform == "linux"