Vivian Liu commited on
Commit
424e416
·
1 Parent(s): 15dab13

with versions

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -1,3 +1,4 @@
 
1
  FROM nvidia/cuda:12.2.0-base-ubuntu22.04
2
 
3
  ENV DEBIAN_FRONTEND noninteractive
@@ -39,7 +40,7 @@ RUN pip3 install --upgrade pip
39
  # Clone the repository and install Python dependencies
40
  RUN git clone https://github.com/hellovivian/generative-disco.git \
41
  && cd generative-disco \
42
- && pip install --ignore-installed stable_diffusion_videos openai numba realesrgan
43
 
44
  # Add user and give permissions
45
  RUN adduser --disabled-password --gecos '' user \
 
1
+
2
  FROM nvidia/cuda:12.2.0-base-ubuntu22.04
3
 
4
  ENV DEBIAN_FRONTEND noninteractive
 
40
  # Clone the repository and install Python dependencies
41
  RUN git clone https://github.com/hellovivian/generative-disco.git \
42
  && cd generative-disco \
43
+ && pip install --ignore-installed stable_diffusion_videos==0.8.0 openai numba realesrgan diffusers=0.11.0
44
 
45
  # Add user and give permissions
46
  RUN adduser --disabled-password --gecos '' user \