Merge pull request dotsimulate#2 from pschroedl/diffusers_module_imports

update paths for diffusers 0.31.0

Author: pschroedl

src/streamdiffusion/acceleration/tensorrt/engine.py

@@ -1,9 +1,10 @@
 from typing import *
 
 import torch
+
+from diffusers.models.unets.unet_2d_condition import UNet2DConditionOutput
 from diffusers.models.autoencoders.autoencoder_tiny import AutoencoderTinyOutput
-from diffusers.models.unet_2d_condition import UNet2DConditionOutput
-from diffusers.models.vae import DecoderOutput
+from diffusers.models.autoencoders.autoencoder_kl import DecoderOutput
 from polygraphy import cuda
 
 from .utilities import Engine