Update ip_adapter/attention_processor.py

ip_adapter/attention_processor.py

@@ -356,8 +356,8 @@ class IPAttnProcessor2_0(torch.nn.Module):
         )
 
         hidden_states = hidden_states.transpose(1, 2).reshape(batch_size, -1, attn.heads * head_dim)
-        #hidden_states = hidden_states.to(query.dtype)
-        hidden_states = hidden_states.to(torch.float32)
+        hidden_states = hidden_states.to(query.dtype)
+        #hidden_states = hidden_states.to(torch.float32)
         
         # for ip-adapter
         ip_key = self.to_k_ip(ip_hidden_states)