NanoBotAIAgent commited on
Commit
7d6ff40
·
verified ·
1 Parent(s): c811320

Upload Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +6 -2
Dockerfile CHANGED
@@ -14,8 +14,12 @@ RUN apt-get update && apt-get install -y \
14
  WORKDIR /app
15
 
16
  # Clone ComfyUI
17
- RUN git clone https://github.com/comfyanonymous/ComfyUI.git . && \
18
- pip install --no-cache-dir torch torchvision --index-url https://download.pytorch.org/whl/cpu && \
 
 
 
 
19
  pip install --no-cache-dir -r requirements.txt
20
 
21
  # Download a small fast model (SD 1.5 pruned EMA-only ~2GB)
 
14
  WORKDIR /app
15
 
16
  # Clone ComfyUI
17
+ RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
18
+
19
+ # Install CPU-only PyTorch stack (torchaudio must come from CPU index or it links libcudart)
20
+ RUN pip install --no-cache-dir \
21
+ torch torchvision torchaudio \
22
+ --index-url https://download.pytorch.org/whl/cpu && \
23
  pip install --no-cache-dir -r requirements.txt
24
 
25
  # Download a small fast model (SD 1.5 pruned EMA-only ~2GB)