redkits commited on
Commit
9b38d05
·
verified ·
1 Parent(s): dcb6cc1

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +20 -0
Dockerfile ADDED
@@ -0,0 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM huggingface/transformers-pytorch-gpu:latest
2
+
3
+ WORKDIR /repository
4
+
5
+ # Install diffusers from source (supports WanPipeline) + flask + imageio for video
6
+ RUN pip install --no-cache-dir
7
+ git+https://github.com/huggingface/diffusers.git
8
+ flask
9
+ accelerate
10
+ sentencepiece
11
+ protobuf
12
+ "imageio[ffmpeg]"
13
+
14
+ COPY server.py /repository/server.py
15
+ COPY handler.py /repository/handler.py
16
+ COPY requirements.txt /repository/requirements.txt
17
+
18
+ EXPOSE 8000
19
+
20
+ CMD ["python", "/repository/server.py"]