| 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"] | |