Skip to content

GitLab

Unverified Commit 723e8f6b authored by Peter's avatar Peter Committed by GitHub
Browse files

Fix ONNX img2img preprocessing (#1736) 


Co-authored-by: default avatarPeter <peterto@users.noreply.github.com>
parent c53a8506
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py
View file @ 723e8f6b
...@@ -332,6 +332,9 @@ class OnnxStableDiffusionImg2ImgPipeline(DiffusionPipeline): ...@@ -332,6 +332,9 @@ class OnnxStableDiffusionImg2ImgPipeline(DiffusionPipeline):
f" {type(callback_steps)}." f" {type(callback_steps)}."
) )
if generator is None:
generator = np.random
# set timesteps # set timesteps
self.scheduler.set_timesteps(num_inference_steps) self.scheduler.set_timesteps(num_inference_steps)
......
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