Fix ONNX img2img preprocessing (huggingface#1736)

peterto · web-flow · commit 723e8f6bb4d6 · 2022-12-17T13:12:10.000+01:00
Co-authored-by: Peter &lt;peterto@users.noreply.github.com&gt;
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`