Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
 # ----------------------------------------------------
-# Dockerfile Comfy UI (Mode CPU) - V5 (FIX PORT 7860)
+# Dockerfile Comfy UI (Mode CPU) - V6 (FINAL)
 # Ce fichier définit l'environnement de votre Hugging Face Space.
 # Il est optimisé pour les instances CPU.
 # ----------------------------------------------------
@@ -10,6 +10,7 @@ FROM python:3.11-slim
 WORKDIR /app
 
 # 2. Installation des dépendances système
+# git pour le clonage, libglib2.0-0 pour Pillow/Torch, curl pour les téléchargements
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     libglib2.0-0 \
@@ -22,17 +23,14 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git
 
 WORKDIR /app/ComfyUI
 
-# 4. Installation des dépendances Python
-# FIX 1 : On garde NumPy < 2.0 pour la compatibilité
+# 4. Installation des dépendances Python (Mode CPU)
+# FIX 1 : NumPy pour la compatibilité
 RUN pip install "numpy<2.0"
-
-# FIX 2 : PyTorch CPU (Essentiel pour éviter l'erreur de driver NVIDIA)
+# FIX 2 : PyTorch CPU pour éviter l'erreur de driver
 RUN pip install torch==2.3.1+cpu torchvision==0.18.1+cpu torchaudio==2.3.1+cpu --index-url https://download.pytorch.org/whl/cpu
-
 # FIX 3 : Dépendances standards
 RUN pip install -r requirements.txt
-
-# FIX 4 : Transformers récent (avec une version de tokenizers compatible)
+# FIX 4 : Transformers récent pour les encodeurs de texte
 RUN pip install transformers==4.41.2 tokenizers==0.19.1
 
 # 5. Téléchargement des modèles SD 1.5 (CHECKPOINTS)
@@ -41,6 +39,12 @@ RUN mkdir -p models/checkpoints \
     && curl -L "https://civitai.com/api/download/models/126689?type=Model&format=SafeTensor&size=full" -o ./models/checkpoints/dreamshaper_8.safetensors \
     && curl -L "https://civitai.com/api/download/models/4201?type=Model&format=SafeTensor&size=full&fp16=false" -o ./models/checkpoints/realisticVisionV51_v51VAE.safetensors
 
+# 5B. AJOUT CRITIQUE : VAEs et CLIP pour éviter les erreurs de génération
+RUN mkdir -p models/vae models/clip \
+    && curl -L "https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.safetensors" -o ./models/vae/vae-ft-mse-840000-ema-pruned.safetensors \
+    && curl -L "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/vae.pt" -o ./models/vae/vae.pt \
+    && curl -L "https://huggingface.co/comfyanonymous/clip_g/resolve/main/clip_g.safetensors" -o ./models/clip/clip_g.safetensors
+
 # 6. Configuration Hugging Face Space (Port 7860 Obligatoire)
 ENV PORT 7860
 EXPOSE 7860