From 4ee0e1d3e7b1bfa9fba229ab9dd8c1aa80afa6e8 Mon Sep 17 00:00:00 2001 From: Krishna Kalyan Date: Sun, 29 Jun 2025 20:18:43 +0000 Subject: [PATCH] tensorrt text2img --- examples/community/stable_diffusion_tensorrt_txt2img.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/community/stable_diffusion_tensorrt_txt2img.py b/examples/community/stable_diffusion_tensorrt_txt2img.py index 15a6e69c413c..1ea41a94ea2a 100755 --- a/examples/community/stable_diffusion_tensorrt_txt2img.py +++ b/examples/community/stable_diffusion_tensorrt_txt2img.py @@ -48,8 +48,8 @@ from diffusers.configuration_utils import FrozenDict, deprecate from diffusers.image_processor import VaeImageProcessor from diffusers.models import AutoencoderKL, UNet2DConditionModel -from diffusers.pipelines.stable_diffusion import ( - StableDiffusionPipelineOutput, +from diffusers.pipelines.stable_diffusion import StableDiffusionPipelineOutput +from diffusers.pipelines.stable_diffusion.safety_checker import ( StableDiffusionSafetyChecker, ) from diffusers.schedulers import DDIMScheduler