xtts: +AMD gpu ROCm, +Apple MPS

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

View file

@ -10,7 +10,11 @@ RUN apt-get clean && rm -rf /var/lib/apt/lists/*
WORKDIR /app
RUN mkdir -p voices config
COPY requirements.txt /app/
ARG USE_ROCM
ENV USE_ROCM=${USE_ROCM}
COPY requirements*.txt /app/
RUN if [ ${USE_ROCM} = "1" ]; then mv /app/requirements-rocm.txt /app/requirements.txt; fi
RUN --mount=type=cache,target=/root/.cache/pip pip install -r requirements.txt
COPY speech.py openedai.py say.py *.sh *.default.yaml README.md LICENSE /app/