Mairaaa commited on
Commit
e22ce03
·
verified ·
1 Parent(s): 8f97965

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -8
Dockerfile CHANGED
@@ -1,18 +1,20 @@
1
  FROM python:3.9-slim
2
 
3
- # Install system dependencies
4
  RUN apt-get update && apt-get install -y \
5
- curl \
6
  build-essential \
7
- cargo \
8
- libsm6 \
9
- libxext6 \
10
- libgl1-mesa-glx \
11
  ffmpeg \
12
- git \
13
- git-lfs \
14
  && rm -rf /var/lib/apt/lists/*
15
 
 
 
 
 
 
 
 
 
 
 
16
  # Set the working directory
17
  WORKDIR /app
18
 
 
1
  FROM python:3.9-slim
2
 
 
3
  RUN apt-get update && apt-get install -y \
 
4
  build-essential \
 
 
 
 
5
  ffmpeg \
 
 
6
  && rm -rf /var/lib/apt/lists/*
7
 
8
+ RUN pip install --no-cache-dir --upgrade pip && \
9
+ pip install --no-cache-dir \
10
+ torch==2.0.1 torchvision==0.15.2 \
11
+ diffusers==0.16.0 \
12
+ accelerate==0.21.0 \
13
+ transformers>=4.26.0 \
14
+ Pillow==9.4.0 \
15
+ opencv-python==4.8.0.76
16
+
17
+
18
  # Set the working directory
19
  WORKDIR /app
20