From 84c403aedb967d68785e6e5ac359746f97a483cb Mon Sep 17 00:00:00 2001 From: Jianqi Pan Date: Tue, 26 Dec 2023 00:46:57 +0900 Subject: [PATCH] fix: cannot set guidance_scale (#6326) 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)