Update Dockerfile

Dockerfile

@@ -1,68 +1,40 @@
 # ----------------------------------------------------
-# Dockerfile Comfy UI (Mode CPU) - V19 (CORRECTION FINALE DU VAE)
-# Correction: Le nom du fichier VAE est incorrect. Utilise maintenant 'vae.safetensors'
-# du dépôt Stability AI (résout le 404 sur le VAE).
+# Dockerfile Comfy UI (Mode GPU) - VERSION FINALE
 # ----------------------------------------------------
 
 # 1. Image de base
 FROM python:3.11-slim
-
-# Définir le répertoire de travail
 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 \
-    wget \
     && rm -rf /var/lib/apt/lists/*
-
-# 3. Installation de ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git
-
 WORKDIR /app/ComfyUI
 
-# 4. Installation des dépendances Python (incluant hf_hub pour téléchargement robuste)
-RUN pip install torch==2.3.1+cpu torchvision==0.18.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
-RUN pip install -r requirements.txt
-RUN pip install huggingface_hub
+# 4. Installation des dépendances Python (CRITIQUE : POUR GPU)
+
+# FIX 1 : On garde NumPy 1.x
+RUN pip install "numpy<2.0"
 
-# 5. Création du répertoire des modèles
-RUN mkdir -p models/checkpoints models/vae
+# FIX 2 : Installation de PyTorch 2.3.1 pour CUDA 11.8 (NÉCESSAIRE POUR GPU)
+# Ceci est l'URL de l'installation GPU.
+RUN pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cu118
 
-# 6. Téléchargement des 2 Modèles (Stable Diffusion 1.5 + VAE)
-RUN echo "import os, glob\n\
-from huggingface_hub import hf_hub_download\n\
-\n\
-# 1. Nettoyage des anciens fichiers\n\
-for f in glob.glob(os.path.join('models', '*', '*.safetensors')):\n\
-    os.remove(f)\n\
-\n\
-# 2. Téléchargement du Checkpoint (Stable Diffusion 1.5 - Public et Non Gated)\n\
-CHECKPOINT_REPO = 'runwayml/stable-diffusion-v1-5'\n\
-CHECKPOINT_FILE = 'v1-5-pruned-emaonly.safetensors'\n\
-CHECKPOINT_DIR = 'models/checkpoints'\n\
-print('--- 1/2 Downloading Checkpoint: {} ---'.format(CHECKPOINT_FILE))\n\
-hf_hub_download(repo_id=CHECKPOINT_REPO, filename=CHECKPOINT_FILE, local_dir=CHECKPOINT_DIR)\n\
-\n\
-# 3. Téléchargement du VAE (VAE officiel de Stability AI - Nom de fichier corrigé)\n\
-VAE_REPO = 'stabilityai/sd-vae-ft-mse'\n\
-VAE_SOURCE_FILE = 'vae.safetensors'\n\
-VAE_DEST_FILE = 'vae.safetensors'\n\
-VAE_DIR = 'models/vae'\n\
-print('--- 2/2 Downloading VAE ({}) ---'.format(VAE_SOURCE_FILE))\n\
-# Utilisation de allow_patterns pour garantir le téléchargement du fichier à la racine\n\
-hf_hub_download(repo_id=VAE_REPO, allow_patterns=VAE_SOURCE_FILE, local_dir=VAE_DIR, local_file_on_download=VAE_DEST_FILE)\n\
-print('Successfully saved VAE as {} in {}'.format(VAE_DEST_FILE, VAE_DIR))\n\
-" > download_models_v19.py
+# FIX 3 : Installation des dépendances standards
+RUN pip install -r requirements.txt
 
-RUN python3 download_models_v19.py
+# FIX 4 : On force une version récente de Transformers
+RUN pip install transformers==4.41.2
 
-# 7. Ajout du noeud pour la sauvegarde d'image par WebSocket
-RUN mkdir -p custom_nodes
-RUN curl -L https://raw.githubusercontent.com/comfyanonymous/ComfyUI/master/custom_nodes/websocket_image_save.py -o custom_nodes/websocket_image_save.py
+# 5. Téléchargement du modèle SD 1.5
+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
 
-# 8. Démarrage de l'application
+# 6. Configuration Hugging Face
+ENV PORT 7860
 EXPOSE 7860
-CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "7860", "--disable-metadata"]
+
+# 7. Lancement (On retire le --cpu car nous sommes en GPU)
+CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "7860"]