tolgacangoz
/

anytext

Model card Files Files and versions

tolgacangoz commited on Feb 24

Commit

b1a65f9

·

verified ·

1 Parent(s): 82a5e8b

Upload anytext.py

Files changed (1) hide show

anytext.py +1 -1

anytext.py CHANGED Viewed

@@ -2335,7 +2335,7 @@ class AnyTextPipeline(
                     cond_scale = controlnet_cond_scale * controlnet_keep[i]
                 down_block_res_samples, mid_block_res_sample = self.controlnet(
-                    control_model_input,
                     t,
                     encoder_hidden_states=controlnet_prompt_embeds,
                     controlnet_cond=guided_hint,

                     cond_scale = controlnet_cond_scale * controlnet_keep[i]
                 down_block_res_samples, mid_block_res_sample = self.controlnet(
+                    control_model_input.to(self.controlnet.dtype),
                     t,
                     encoder_hidden_states=controlnet_prompt_embeds,
                     controlnet_cond=guided_hint,