Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +14 -17
Dockerfile
CHANGED
|
@@ -1,19 +1,21 @@
|
|
| 1 |
-
FROM python:3.11
|
| 2 |
|
| 3 |
-
#
|
| 4 |
-
WORKDIR /home/user/app
|
| 5 |
-
ENV HF_HOME="/home/user/.cache/huggingface"
|
| 6 |
-
ENV PYTHONUNBUFFERED=1
|
| 7 |
-
ENV NUMBA_CACHE_DIR=/tmp/numba_cache
|
| 8 |
-
ENV NUMBA_DISABLE_CACHING=1
|
| 9 |
-
ENV OMP_NUM_THREADS=1
|
| 10 |
-
|
| 11 |
-
# Instalar dependencias del sistema
|
| 12 |
RUN apt-get update && apt-get install -y \
|
| 13 |
git wget ffmpeg libsndfile1 unzip build-essential \
|
| 14 |
&& rm -rf /var/lib/apt/lists/*
|
| 15 |
|
| 16 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
RUN git clone --branch v0.22.0 https://github.com/coqui-ai/TTS.git /home/user/TTS \
|
| 18 |
&& cd /home/user/TTS \
|
| 19 |
&& git submodule update --init --recursive \
|
|
@@ -24,12 +26,7 @@ RUN git clone --branch v0.22.0 https://github.com/coqui-ai/TTS.git /home/user/TT
|
|
| 24 |
COPY requirements.txt .
|
| 25 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 26 |
|
| 27 |
-
#
|
| 28 |
-
RUN mkdir -p /tmp/output_model
|
| 29 |
-
|
| 30 |
-
# Copiar el proyecto
|
| 31 |
COPY . .
|
| 32 |
|
| 33 |
-
# Script de entrenamiento
|
| 34 |
CMD ["python", "finetune_xtts_hf.py"]
|
| 35 |
-
|
|
|
|
| 1 |
+
FROM python:3.11
|
| 2 |
|
| 3 |
+
# Instalar dependencias básicas
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
RUN apt-get update && apt-get install -y \
|
| 5 |
git wget ffmpeg libsndfile1 unzip build-essential \
|
| 6 |
&& rm -rf /var/lib/apt/lists/*
|
| 7 |
|
| 8 |
+
# Configurar Hugging Face cache y variables de entorno
|
| 9 |
+
ENV HF_HOME=/home/user/.cache/huggingface
|
| 10 |
+
ENV OMP_NUM_THREADS=1
|
| 11 |
+
ENV NUMBA_CACHE_DIR=/tmp/numba_cache
|
| 12 |
+
ENV NUMBA_DISABLE_CACHING=1
|
| 13 |
+
|
| 14 |
+
# Crear carpetas necesarias
|
| 15 |
+
RUN mkdir -p /home/user/app /tmp/output_model ${HF_HOME}
|
| 16 |
+
WORKDIR /home/user/app
|
| 17 |
+
|
| 18 |
+
# Instalar Coqui TTS desde GitHub (incluyendo binarios de entrenamiento)
|
| 19 |
RUN git clone --branch v0.22.0 https://github.com/coqui-ai/TTS.git /home/user/TTS \
|
| 20 |
&& cd /home/user/TTS \
|
| 21 |
&& git submodule update --init --recursive \
|
|
|
|
| 26 |
COPY requirements.txt .
|
| 27 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 28 |
|
| 29 |
+
# Copiar el script de fine-tuning y dataset
|
|
|
|
|
|
|
|
|
|
| 30 |
COPY . .
|
| 31 |
|
|
|
|
| 32 |
CMD ["python", "finetune_xtts_hf.py"]
|
|
|