dieumercimvemba commited on
Commit
80b207b
·
verified ·
1 Parent(s): 04e7d99

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +20 -16
Dockerfile CHANGED
@@ -1,12 +1,14 @@
1
- # ----------------------------------------------------# Dockerfile Comfy UI (Mode CPU pour Hugging Face Spaces) - V3 (FIX final de version)# ----------------------------------------------------
 
 
2
 
3
- # 1. Image de base (Python 3.11)
4
  FROM python:3.11-slim
5
 
6
  # Définir le répertoire de travail
7
  WORKDIR /app
8
 
9
- # 2. Installation des dépendances système nécessaires
10
  RUN apt-get update && apt-get install -y --no-install-recommends \
11
  git \
12
  libglib2.0-0 \
@@ -16,28 +18,30 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
16
  # 3. Clonage de ComfyUI
17
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git
18
 
19
- # 4. Installation des dépendances Python (CRITIQUE : FORCER CPU ET FIX DE VERSION)
20
  WORKDIR /app/ComfyUI
21
 
22
- # FIX 1: Downgrade de NumPy (pour résoudre les avertissements/crash potentiels)
23
- RUN pip install numpy==1.26.4
24
 
25
- # FIX 2: Installation de PyTorch pour CPU
26
- 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
 
27
 
28
- # FIX 3 (NOUVEAU): Installation des autres librairies NECESSAIRES, y compris le downgrade de transformers
29
- # Nous installons les dépendances en ignorant la version de transformers du requirements.txt
30
- # Puis nous installons une version de transformers connue pour être compatible avec PyTorch 2.1.0.
31
- RUN pip install --no-deps -r requirements.txt
32
- RUN pip install transformers==4.35.2
33
 
34
- # 5. Téléchargement d'un modèle (Requis pour démarrer)
 
 
 
 
35
  RUN mkdir -p models/checkpoints \
36
  && 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
37
 
38
- # 6. Configuration Hugging Face Space
39
  ENV PORT 7860
40
  EXPOSE 7860
41
 
42
- # 7. Lancement de ComfyUI (CRITIQUE : FORCER CPU)
43
  CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"]
 
1
+ # ----------------------------------------------------
2
+ # Dockerfile Comfy UI (Mode CPU) - V4 (Upgrade Stack)
3
+ # ----------------------------------------------------
4
 
5
+ # 1. Image de base
6
  FROM python:3.11-slim
7
 
8
  # Définir le répertoire de travail
9
  WORKDIR /app
10
 
11
+ # 2. Installation des dépendances système
12
  RUN apt-get update && apt-get install -y --no-install-recommends \
13
  git \
14
  libglib2.0-0 \
 
18
  # 3. Clonage de ComfyUI
19
  RUN git clone https://github.com/comfyanonymous/ComfyUI.git
20
 
21
+ # 4. Installation des dépendances Python
22
  WORKDIR /app/ComfyUI
23
 
24
+ # FIX 1 : On garde NumPy 1.x pour éviter les crashs de la v2.0
25
+ RUN pip install "numpy<2.0"
26
 
27
+ # FIX 2 (UPGRADE) : On passe à PyTorch 2.3.1 (CPU)
28
+ # Cela permet de supporter les versions récentes de Transformers
29
+ 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
30
 
31
+ # FIX 3 : Installation des dépendances standards
32
+ RUN pip install -r requirements.txt
 
 
 
33
 
34
+ # FIX 4 : On force une version récente de Transformers qui supporte Qwen2Tokenizer
35
+ # PyTorch 2.3.1 est compatible avec cette version.
36
+ RUN pip install transformers==4.41.2
37
+
38
+ # 5. Téléchargement du modèle SD 1.5
39
  RUN mkdir -p models/checkpoints \
40
  && 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
41
 
42
+ # 6. Configuration Hugging Face
43
  ENV PORT 7860
44
  EXPOSE 7860
45
 
46
+ # 7. Lancement
47
  CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"]