apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix controlnet with fp16 again

This commit is contained in:
Sean Sube 2023-12-19 23:16:19 -06:00
parent e09631ef05
commit 9c215dd90b
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/onnx_web/diffusers/pipelines/controlnet.py
View File

@ -416,9 +416,9 @@ class OnnxStableDiffusionControlNetPipeline(DiffusionPipeline):
timestep = np.array([t], dtype=timestep_dtype) timestep = np.array([t], dtype=timestep_dtype)
blocksamples = self.controlnet( blocksamples = self.controlnet(
sample=latent_model_input.astype(np.float32), sample=latent_model_input,
timestep=timestep, timestep=timestep,
encoder_hidden_states=prompt_embeds.astype(np.float32), encoder_hidden_states=prompt_embeds,
controlnet_cond=image, controlnet_cond=image,
) )