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