Update Dockerfile

Dockerfile

@@ -1,33 +1,49 @@
 # ----------------------------------------------------
 # Dockerfile Comfy UI (Mode CPU) - V5 (FIX PORT 7860)
+# Ce fichier définit l'environnement de votre Hugging Face Space.
+# Il est optimisé pour les instances CPU.
 # ----------------------------------------------------
 
 # 1. Image de base
-# 4. Installation des dépendances Python
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# 2. Installation des dépendances système
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    libglib2.0-0 \
+    curl \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
+
+# 3. Clonage de ComfyUI
+RUN git clone https://github.com/comfyanonymous/ComfyUI.git
+
 WORKDIR /app/ComfyUI
 
-# FIX 1 : On garde NumPy 1.x pour la stabilité
+# 4. Installation des dépendances Python
+# FIX 1 : On garde NumPy < 2.0 pour la compatibilité
 RUN pip install "numpy<2.0"
 
-# FIX 2 : PyTorch CPU (Mise à jour pour supporter Transformers récent)
-
+# FIX 2 : PyTorch CPU (Essentiel pour éviter l'erreur de driver NVIDIA)
 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 pour Qwen/Hunyuan
-
-RUN pip install transformers==4.41.2
+# FIX 4 : Transformers récent (avec une version de tokenizers compatible)
+RUN pip install transformers==4.41.2 tokenizers==0.19.1
 
-# 5. Téléchargement du modèle SD 1.5
+# 5. Téléchargement des modèles SD 1.5 (CHECKPOINTS)
 RUN mkdir -p models/checkpoints \
-    && curl -L "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors" -o ./models/checkpoints/v1-5-pruned-emaonly.safetensors
+    && curl -L "https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.safetensors" -o ./models/checkpoints/v1-5-pruned-emaonly.safetensors \
+    && 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
 
 # 6. Configuration Hugging Face Space (Port 7860 Obligatoire)
 ENV PORT 7860
 EXPOSE 7860
 
-# 7. Lancement (CRITIQUE : PORT 7860)
-# On force ComfyUI à écouter sur le port 7860 au lieu de 8188
+# 7. Lancement (CRITIQUE : PORT 7860 et mode --cpu)
 CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--cpu"]