Merge branch 'main' of github.com:ostris/ai-toolkit

2024-06-29 15:53:40 -06:00
parent 3072d20f17 f6b21f47bb
commit bb57623a35
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