From 20bae90995550f3e55fa0147aa8ee6c0cb7fbfa7 Mon Sep 17 00:00:00 2001 From: Jianqi Pan Date: Mon, 25 Dec 2023 22:54:13 +0900 Subject: [PATCH] fix: set guidance_scale --- examples/community/stable_diffusion_tensorrt_img2img.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/community/stable_diffusion_tensorrt_img2img.py b/examples/community/stable_diffusion_tensorrt_img2img.py index e6e5e9db71d0..a391daf1062d 100755 --- a/examples/community/stable_diffusion_tensorrt_img2img.py +++ b/examples/community/stable_diffusion_tensorrt_img2img.py @@ -1004,7 +1004,7 @@ def __call__( """ self.generator = generator self.denoising_steps = num_inference_steps - self.guidance_scale = guidance_scale + self._guidance_scale = guidance_scale # Pre-compute latent input scales and linear multistep coefficients self.scheduler.set_timesteps(self.denoising_steps, device=self.torch_device)