sob111 commited on
Commit
557d2bd
·
verified ·
1 Parent(s): 393bff8

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +14 -17
Dockerfile CHANGED
@@ -1,19 +1,21 @@
1
- FROM python:3.11-slim
2
 
3
- # Configurar paths
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
- # Clonar e instalar Coqui TTS v0.22 en modo editable (con train y recipes)
 
 
 
 
 
 
 
 
 
 
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
- # Crear carpeta de salida
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"]