Fix ONNX img2img preprocessing

peterto · peterto · commit 944d149d9231 · 2022-12-16T16:45:21.000-05:00
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py b/src/diffusers/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py
@@ -332,6 +332,9 @@ def __call__(
                 f" {type(callback_steps)}."
             )
 
+        if generator is None:
+            generator = np.random
+
         # set timesteps
         self.scheduler.set_timesteps(num_inference_steps)
 

Original file line number	Diff line number	Diff line change
`@@ -332,6 +332,9 @@ def __call__(`
`332`	`332`	`f" {type(callback_steps)}."`
`333`	`333`	`)`
`334`	`334`
	`335`	`+ if generator is None:`
	`336`	`+ generator = np.random`
	`337`	`+`
`335`	`338`	`# set timesteps`
`336`	`339`	`self.scheduler.set_timesteps(num_inference_steps)`
`337`	`340`