File tree 1 file changed +1
-7
lines changed
1 file changed +1
-7
lines changed Original file line number Diff line number Diff line change @@ -4,12 +4,6 @@ FROM runpod/base:0.6.1-cuda12.2.0
4
4
# Only install absolutely necessary packages for your project
5
5
RUN apt-get update && apt-get install -y ffmpeg aria2 git unzip
6
6
7
- # Install Python dependencies
8
- RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 && \
9
- pip install --no-cache-dir opencv-python imageio imageio-ffmpeg ffmpeg-python av runpod \
10
- xformers==0.0.25 torchsde==0.2.6 einops==0.8.0 diffusers==0.28.0 transformers==4.41.2 accelerate==0.30.1
11
-
12
-
13
7
# Clone the ComfyUI repository
14
8
RUN git clone https://github.com/comfyanonymous/ComfyUI /content/ComfyUI
15
9
@@ -28,4 +22,4 @@ RUN mkdir -p /content/ComfyUI/models/unet && \
28
22
WORKDIR /
29
23
30
24
# Keep the base image's entrypoint
31
- ENTRYPOINT ["/start.sh" ]
25
+ CMD ["/start.sh" ]
You can’t perform that action at this time.
0 commit comments