fix controlnet with fp16 again
This commit is contained in:
parent
e09631ef05
commit
9c215dd90b
|
@ -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,
|
||||
)
|
||||
|
||||
|
|
Loading…
Reference in New Issue