fix: cannot set guidance_scale (#6326)

fix: set guidance_scale
huggingface · Dec 25, 2023 · 84c403a · 84c403a
1 parent f4b0b26
commit 84c403a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)