redkits commited on
Commit
164703b
·
verified ·
1 Parent(s): 2c87bcd

Upload Dockerfile with huggingface_hub

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