ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-27 17:51:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Work to omprove pixart training

Browse Source
This commit is contained in:
Jaret Burkett
2024-06-23 20:46:48 +00:00
parent 5d47244c57
commit 7165f2d25a
6 changed files with 65 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/ip_adapter.py
View File

@@ -333,7 +333,7 @@ class IPAdapter(torch.nn.Module):
if not self.config.train_image_encoder:
# compile it
print('Compiling image encoder')
torch.compile(self.image_encoder, fullgraph=True)
#torch.compile(self.image_encoder, fullgraph=True)
self.input_size = self.image_encoder.config.image_size