Increased the number of heads for ip adapters.

2026-02-27 15:53:58 +00:00 · 2024-06-28 16:09:52 +00:00
parent 603ceca3ca
commit f6b21f47bb
1 changed files with 1 additions and 1 deletions
--- 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