File size: 508 Bytes
9b38d05 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | FROM huggingface/transformers-pytorch-gpu:latest
WORKDIR /repository
# Install diffusers from source (supports WanPipeline) + flask + imageio for video
RUN pip install --no-cache-dir
git+https://github.com/huggingface/diffusers.git
flask
accelerate
sentencepiece
protobuf
"imageio[ffmpeg]"
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"]
|