Skip to content

GitLab

Unverified Commit c09bb588 authored by caiyueliang's avatar caiyueliang Committed by GitHub
Browse files

fix: TensorRTStableDiffusionPipeline cannot set guidance_scale (#7065)

parent 66a7160f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/community/stable_diffusion_tensorrt_inpaint.py
View file @ c09bb588
...@@ -1011,7 +1011,7 @@ class TensorRTStableDiffusionInpaintPipeline(StableDiffusionInpaintPipeline): ...@@ -1011,7 +1011,7 @@ class TensorRTStableDiffusionInpaintPipeline(StableDiffusionInpaintPipeline):
""" """
self.generator = generator self.generator = generator
self.denoising_steps = num_inference_steps 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 # Pre-compute latent input scales and linear multistep coefficients
self.scheduler.set_timesteps(self.denoising_steps, device=self.torch_device) self.scheduler.set_timesteps(self.denoising_steps, device=self.torch_device)
......
examples/community/stable_diffusion_tensorrt_txt2img.py
View file @ c09bb588
...@@ -882,7 +882,7 @@ class TensorRTStableDiffusionPipeline(StableDiffusionPipeline): ...@@ -882,7 +882,7 @@ class TensorRTStableDiffusionPipeline(StableDiffusionPipeline):
""" """
self.generator = generator self.generator = generator
self.denoising_steps = num_inference_steps 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 # Pre-compute latent input scales and linear multistep coefficients
self.scheduler.set_timesteps(self.denoising_steps, device=self.torch_device) self.scheduler.set_timesteps(self.denoising_steps, device=self.torch_device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment