Work to omprove pixart training

This commit is contained in:
Jaret Burkett committed 2024-06-23 20:46:48 +00:00
1 parent 5d47244c57
commit 7165f2d25a
6 files changed
+65 -18

No files matched your search

+1 -1
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