dieumercimvemba commited on
Commit
5db6b40
·
verified ·
1 Parent(s): 17ee3dc

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +8 -7
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- # ----------------------------------------------------# Dockerfile Comfy UI (Mode CPU pour Hugging Face Spaces)# ----------------------------------------------------
2
 
3
  # 1. Image de base (Python 3.11)
4
  FROM python:3.11-slim
@@ -10,7 +10,6 @@ WORKDIR /app
10
  RUN apt-get update && apt-get install -y --no-install-recommends \
11
  git \
12
  libglib2.0-0 \
13
- # AJOUT CRITIQUE : curl pour le téléchargement du modèle
14
  curl \
15
  # Nettoyage
16
  && rm -rf /var/lib/apt/lists/*
@@ -18,17 +17,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
18
  # 3. Clonage de ComfyUI
19
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git
20
 
21
- # 4. Installation des dépendances Python (CRITIQUE : FORCER CPU)
22
  WORKDIR /app/ComfyUI
23
 
24
- # Installation de PyTorch et Torchvision, FORCÉMENT pour CPU
 
 
 
 
25
  RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu torchaudio==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu
26
 
27
- # Installation des autres librairies nécessaires à ComfyUI
28
  RUN pip install -r requirements.txt
29
 
30
  # 5. Téléchargement d'un modèle (Requis pour démarrer)
31
- # Modèle Stable Diffusion v1.5 (léger, adapté au CPU pour les tests)
32
  RUN mkdir -p models/checkpoints \
33
  && 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
34
 
@@ -37,5 +39,4 @@ ENV PORT 7860
37
  EXPOSE 7860
38
 
39
  # 7. Lancement de ComfyUI (CRITIQUE : FORCER CPU)
40
- # Écoute sur 0.0.0.0 et utilise le mode CPU
41
  CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"]
 
1
+ # ----------------------------------------------------# Dockerfile Comfy UI (Mode CPU pour Hugging Face Spaces) - V2 (FIX de version)# ----------------------------------------------------
2
 
3
  # 1. Image de base (Python 3.11)
4
  FROM python:3.11-slim
 
10
  RUN apt-get update && apt-get install -y --no-install-recommends \
11
  git \
12
  libglib2.0-0 \
 
13
  curl \
14
  # Nettoyage
15
  && rm -rf /var/lib/apt/lists/*
 
17
  # 3. Clonage de ComfyUI
18
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git
19
 
20
+ # 4. Installation des dépendances Python (CRITIQUE : FORCER CPU ET FIX DE VERSION)
21
  WORKDIR /app/ComfyUI
22
 
23
+ # FIX 1: Downgrade de NumPy à une version 1.x stable pour éviter les conflits
24
+ RUN pip install numpy==1.26.4
25
+
26
+ # FIX 2: Installation de PyTorch pour CPU
27
+ # Nous utilisons la même version 2.1.0+cpu, car les autres dépendances sont corrigées ci-dessus.
28
  RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu torchaudio==2.1.0+cpu --index-url https://download.pytorch.org/whl/cpu
29
 
30
+ # FIX 3: Installation des autres librairies nécessaires à ComfyUI (y compris Transformers)
31
  RUN pip install -r requirements.txt
32
 
33
  # 5. Téléchargement d'un modèle (Requis pour démarrer)
 
34
  RUN mkdir -p models/checkpoints \
35
  && 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
36
 
 
39
  EXPOSE 7860
40
 
41
  # 7. Lancement de ComfyUI (CRITIQUE : FORCER CPU)
 
42
  CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"]