mathieu/ai-suite-rocm
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update xtts

Browse Source
This commit is contained in:
Mathieu Broillet 2024-02-17 11:06:25 +01:00
parent 7803eed885
commit 43f354e902
Signed by: mathieu
GPG Key ID: C0E9E0E95AF03319
2 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -121,6 +121,9 @@ RUN python3.10 -m venv /ai/venv/xtts/ --system-site-packages
RUN sed -i '/torch==2.1.1/d' requirements.txt
RUN sed -i '/torchaudio==2.1.1/d' requirements.txt
RUN /ai/venv/xtts/bin/python -m pip install -r requirements.txt
RUN sed -i 's/device = "cuda" if torch.cuda.is_available() else "cpu"/device = "cpu"/' /ai/xtts/scripts/utils/formatter.py # disable gpu for faster-whipser as ROCM isn't supported yet
RUN sed -i 's/asr_model = WhisperModel(whisper_model, device=device, compute_type="float16")/asr_model = WhisperModel(whisper_model, device=device, compute_type="int8")/' /ai/xtts/scripts/utils/formatter.py
# Install deepspeed and ninja even though it doesn't work with XTTS (dunno why, compilation errors)
RUN /ai/venv/xtts/bin/python -m pip install deepspeed ninja
RUN apt-get install -y ninja-build

2
docker-compose.yml
View File

@ -143,3 +143,5 @@ services:
volumes:
- ./xtts/outputs:/ai/xtts/output
- ./xtts/models:/ai/xtts/models
- ./xtts/hfmodels:/root/.cache/huggingface
# - ./xtts/finetuned_models:/ai/xtts/finetuned_models