aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-19 17:54:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

handle open clip adapter image size being a tuple

Browse Source
This commit is contained in:
Phil Wang
2022-09-19 10:27:09 -07:00
parent 86e2d5ba84
commit d0c11b30b0
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -406,7 +406,10 @@ class OpenClipAdapter(BaseClipAdapter):
@property
def image_size(self):
return self.clip.visual.image_size
image_size = self.clip.visual.image_size
if isinstance(image_size, tuple):
return max(image_size)
return image_size
@property
def image_channels(self):