Update Dockerfile

Dockerfile

@@ -1,5 +1,7 @@
 # ----------------------------------------------------
 # 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
@@ -9,6 +11,7 @@ FROM python:3.11-slim
 WORKDIR /app
 
 # 2. Installation des dépendances système
+# Ajout de git, libglib2.0-0 et curl
 RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     libglib2.0-0 \
@@ -21,26 +24,29 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git
 # 4. Installation des dépendances Python
 WORKDIR /app/ComfyUI
 
-# FIX 1 : On garde NumPy 1.x pour la stabilité
+# FIX 1 : On garde NumPy 1.x pour la stabilité et pour éviter les problèmes de PyTorch
 RUN pip install "numpy<2.0"
 
-# FIX 2 : PyTorch CPU (Mise à jour pour supporter Transformers récent)
+# FIX 2 : PyTorch CPU (Mise à jour essentielle pour l'inférence sur CPU)
+# C'est la version de PyTorch compilée spécifiquement pour le CPU, sans dépendance CUDA/GPU.
 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
+# FIX 3 : Dépendances standards de ComfyUI
 RUN pip install -r requirements.txt
 
-# FIX 4 : Transformers récent pour Qwen/Hunyuan
+# FIX 4 : Transformers récent (utile si vous utilisez des LoRAs ou d'autres modèles d'encodeur)
 RUN pip install transformers==4.41.2
 
-# 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)
+# Le paramètre --cpu force l'utilisation exclusive du CPU, ce qui est nécessaire sur votre instance.
 CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--cpu"]