dieumercimvemba commited on
Commit
a0fb44a
·
verified ·
1 Parent(s): 5656c04

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +39 -0
Dockerfile ADDED
@@ -0,0 +1,39 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
5
+ WORKDIR /app
6
+
7
+ # 2. Installation des dépendances système nécessaires
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
+
14
+ # 3. Clonage de ComfyUI
15
+ RUN git clone https://github.com/comfyanonymous/ComfyUI.git
16
+
17
+ # 4. Installation des dépendances Python (CRITIQUE : FORCER CPU)
18
+ WORKDIR /app/ComfyUI
19
+
20
+ # Installation de PyTorch et Torchvision, FORCÉMENT pour CPU
21
+ 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
22
+
23
+ # Installation des autres librairies nécessaires à ComfyUI
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
+
33
+ # 6. Configuration Hugging Face Space
34
+ 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"]