ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 19:21:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Increased the number of heads for ip adapters.

Browse Source
This commit is contained in:
Jaret Burkett
2024-06-28 16:09:52 +00:00
parent 603ceca3ca
commit f6b21f47bb
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/ip_adapter.py
View File

@@ -410,7 +410,7 @@ class IPAdapter(torch.nn.Module):
if is_pixart: if is_pixart:
# heads = 20 # heads = 20
heads = 12 heads = 20
# dim = 4096 # dim = 4096
dim = 1280 dim = 1280
output_dim = 4096 output_dim = 4096