Work to omprove pixart training
This commit is contained in:
1 parent
5d47244c57
commit
7165f2d25a
6 files changed
+65
-18
No files matched your search
@@ -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
|
||||
|
||||
|
||||
Reference in new issue
Block a user