Skip to content

Commit

Permalink
runner/docker: Fix streamdiffusion deps installation (#277)
Browse files Browse the repository at this point in the history
The diffusers and protobuf should come from the regular pypy repo,
not the pytorch index.
  • Loading branch information
victorges authored Nov 14, 2024
1 parent 4f90470 commit d75e3e0
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions runner/docker/Dockerfile.live-base-streamdiffusion
Original file line number Diff line number Diff line change
Expand Up @@ -16,12 +16,12 @@ RUN pip install --no-cache-dir --upgrade pip==${PIP_VERSION} setuptools==69.5.1
RUN pip install --no-cache-dir \
torch==2.1.0 \
torchvision==0.16.0 \
diffusers==0.30.0 \

xformers \
protobuf==5.27.2 \
--index-url https://download.pytorch.org/whl/cu121
RUN pip install huggingface-hub==0.23.2
RUN pip install --no-cache-dir \
huggingface-hub==0.23.2 \
diffusers==0.30.0 \
protobuf==5.27.2

# Install StreamDiffusion @ 765d71029b1404b94aee2865178d71c257c20318 (latest at time of writing)
RUN pip install git+https://github.com/cumulo-autumn/StreamDiffusion.git@765d710#egg=streamdiffusion[tensorrt]
Expand Down

0 comments on commit d75e3e0

Please sign in to comment.