mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 17:54:20 +01:00
handle open clip adapter image size being a tuple
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -1 +1 @@
|
||||
__version__ = '1.10.5'
|
||||
__version__ = '1.10.6'
|
||||
|
||||
Reference in New Issue
Block a user