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

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -5
Dockerfile CHANGED
@@ -1,4 +1,6 @@
1
- # 1. Image de base (Python pour les dépendances d'IA)
 
 
2
  FROM python:3.11-slim
3
 
4
  # Définir le répertoire de travail
@@ -8,6 +10,8 @@ WORKDIR /app
8
  RUN apt-get update && apt-get install -y --no-install-recommends \
9
  git \
10
  libglib2.0-0 \
 
 
11
  # Nettoyage
12
  && rm -rf /var/lib/apt/lists/*
13
 
@@ -24,9 +28,7 @@ RUN pip install torch==2.1.0+cpu torchvision==0.16.0+cpu torchaudio==2.1.0+cpu -
24
  RUN pip install -r requirements.txt
25
 
26
  # 5. Téléchargement d'un modèle (Requis pour démarrer)
27
- # REMPLACER L'URL par l'URL d'un checkpoint Stable Diffusion 1.5 pour CPU.
28
- # Cette ligne est cruciale pour que ComfyUI ait un modèle à charger au démarrage.
29
- # Exemple pour SD 1.5 :
30
  RUN mkdir -p models/checkpoints \
31
  && 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
32
 
@@ -35,5 +37,5 @@ ENV PORT 7860
35
  EXPOSE 7860
36
 
37
  # 7. Lancement de ComfyUI (CRITIQUE : FORCER CPU)
38
- # Utilise le port 8188 de ComfyUI, mais l'expose sur le 7860 pour Hugging Face Spaces.
39
  CMD ["python3", "main.py", "--listen", "0.0.0.0", "--port", "8188", "--cpu"]
 
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
5
 
6
  # Définir le répertoire de travail
 
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/*
17
 
 
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
  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"]