From f6b21f47bbdd7202ea144c497144c996eafc92f0 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Fri, 28 Jun 2024 16:09:52 +0000
Subject: [PATCH] Increased the number of heads for ip adapters.

---
 toolkit/ip_adapter.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/toolkit/ip_adapter.py b/toolkit/ip_adapter.py
index 4e91e295..9104aac8 100644
--- a/toolkit/ip_adapter.py
+++ b/toolkit/ip_adapter.py
@@ -410,7 +410,7 @@ class IPAdapter(torch.nn.Module):
 
             if is_pixart:
                 # heads = 20
-                heads = 12
+                heads = 20
                 # dim = 4096
                 dim = 1280
                 output_dim = 4096