fix(api): handle mixed tensors for ControlNet on CPU

ssube · Apr 15, 2023 · 3196535 · 3196535
1 parent 161913b
commit 3196535
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/api/onnx_web/diffusers/pipelines/controlnet.py b/api/onnx_web/diffusers/pipelines/controlnet.py
@@ -169,7 +169,11 @@ def prepare_latents(
             latents = generator.randn(*shape).astype(dtype)
 
         # scale the initial noise by the standard deviation required by the scheduler
-        latents = latents * self.scheduler.init_noise_sigma
+        sigma = self.scheduler.init_noise_sigma
+        if torch.is_tensor(sigma):
+            sigma = sigma.numpy()
+
+        latents = latents * sigma
         return latents
 
     # Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.StableDiffusionPipeline.prepare_extra_step_kwargs