From 9c215dd90b4e6a24b1e75698f486ccc68619925b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 19 Dec 2023 23:16:19 -0600 Subject: [PATCH] fix controlnet with fp16 again --- api/onnx_web/diffusers/pipelines/controlnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/diffusers/pipelines/controlnet.py b/api/onnx_web/diffusers/pipelines/controlnet.py index 5d7bad58..be16265f 100644 --- a/api/onnx_web/diffusers/pipelines/controlnet.py +++ b/api/onnx_web/diffusers/pipelines/controlnet.py @@ -416,9 +416,9 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline): timestep = np.array([t], dtype=timestep_dtype) blocksamples = self.controlnet( - sample=latent_model_input.astype(np.float32), + sample=latent_model_input, timestep=timestep, - encoder_hidden_states=prompt_embeds.astype(np.float32), + encoder_hidden_states=prompt_embeds, controlnet_cond=image, )