3

Dockerfile

@@ -8,7 +8,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 # Instalar dependencias de Python específicas para Marigold
-RUN pip install --no-cache-dir diffusers==0.16.1 transformers==4.30.2 huggingface_hub==0.16.4 fastapi uvicorn torch torchvision pillow accelerate
+RUN pip install --no-cache-dir diffusers==0.16.1 transformers==4.30.2 huggingface_hub==0.16.4 accelerate==0.21.0 fastapi uvicorn torch torchvision pillow
+
 
 # Crear directorio de caché con permisos
 RUN mkdir -p /app/cache && chmod -R 777 /app/cache