From c3a66b016ba751c84920d7651f4240cebfce2185 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Wed, 7 Feb 2024 12:39:55 -0800 Subject: [PATCH] try solve dtype cast for #112 --- .../sd_forge_ipadapter/lib_ipadapter/IPAdapterPlus.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/extensions-builtin/sd_forge_ipadapter/lib_ipadapter/IPAdapterPlus.py b/extensions-builtin/sd_forge_ipadapter/lib_ipadapter/IPAdapterPlus.py index ea7743c0..1ea65022 100644 --- a/extensions-builtin/sd_forge_ipadapter/lib_ipadapter/IPAdapterPlus.py +++ b/extensions-builtin/sd_forge_ipadapter/lib_ipadapter/IPAdapterPlus.py @@ -463,7 +463,7 @@ class CrossAttentionPatch: ip_k = ip_k * W ip_v = ip_v_offset + ip_v_mean * W - out_ip = optimized_attention(q, ip_k, ip_v, extra_options["n_heads"]) + out_ip = optimized_attention(q, ip_k.to(org_dtype), ip_v.to(org_dtype), extra_options["n_heads"]) if weight_type.startswith("original"): out_ip = out_ip * weight