From 01d6d6eefb9b8f0a62c19ca26835f3252428f970 Mon Sep 17 00:00:00 2001 From: flecavalier Date: Tue, 4 Mar 2025 10:20:31 -0500 Subject: [PATCH 1/3] Update diffusers --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index baa55903..d8cc9634 100644 --- a/setup.py +++ b/setup.py @@ -7,7 +7,7 @@ _deps = [ "torch", "xformers", - "diffusers==0.24.0", + "diffusers", "transformers", "accelerate", "fire", From 88f5a165e3bcd2fa87305e791f4999eb4ae13f4f Mon Sep 17 00:00:00 2001 From: flecavalier Date: Tue, 4 Mar 2025 10:20:43 -0500 Subject: [PATCH 2/3] Fix tensorrt imports --- src/streamdiffusion/acceleration/tensorrt/engine.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/streamdiffusion/acceleration/tensorrt/engine.py b/src/streamdiffusion/acceleration/tensorrt/engine.py index 43d65247..cff8ecc7 100644 --- a/src/streamdiffusion/acceleration/tensorrt/engine.py +++ b/src/streamdiffusion/acceleration/tensorrt/engine.py @@ -1,9 +1,9 @@ from typing import * import torch -from diffusers.models.autoencoder_tiny import AutoencoderTinyOutput -from diffusers.models.unet_2d_condition import UNet2DConditionOutput -from diffusers.models.vae import DecoderOutput +from diffusers.models.autoencoders.autoencoder_tiny import AutoencoderTinyOutput +from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput +from diffusers.models.autoencoders.vae import DecoderOutput from polygraphy import cuda from .utilities import Engine From acb5370b022b45d35af149151bb87cb8b406b1e8 Mon Sep 17 00:00:00 2001 From: flecavalier Date: Tue, 4 Mar 2025 10:20:57 -0500 Subject: [PATCH 3/3] Pin numpy to 1.26.4 in Dockerfile --- Dockerfile | 2 ++ 1 file changed, 2 insertions(+) diff --git a/Dockerfile b/Dockerfile index 7aa38bf9..d0eb39bd 100644 --- a/Dockerfile +++ b/Dockerfile @@ -29,6 +29,8 @@ RUN pip3 install \ xformers \ --index-url https://download.pytorch.org/whl/cu118 +RUN pip install numpy==1.26.4 + COPY . /streamdiffusion WORKDIR /streamdiffusion