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