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

13
requirements.txt
View file

@ -5,8 +5,15 @@ loguru
piper-tts==1.2.0
# xtts
TTS
# Fixes: https://github.com/matatonic/openedai-speech/issues/9
# Re: https://github.com/pytorch/pytorch/issues/121834
torch==2.2.2
# 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; sys_platform != "darwin"
torchaudio; sys_platform != "darwin"
# for MPS accelerated torch on Mac
torch==2.2.2; --index-url https://download.pytorch.org/whl/cpu; sys_platform == "darwin"
torchaudio==2.2.2; --index-url https://download.pytorch.org/whl/cpu; sys_platform == "darwin"
# ROCM (Linux only) - use requirements.amd.txt