mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2025-12-19 09:44:19 +01:00
protect against random cropping for base unet
This commit is contained in:
@@ -2172,6 +2172,7 @@ class Decoder(nn.Module):
|
||||
# random crop sizes (for super-resoluting unets at the end of cascade?)
|
||||
|
||||
self.random_crop_sizes = cast_tuple(random_crop_sizes, len(image_sizes))
|
||||
assert not exists(self.random_crop_sizes[0]), 'you would not need to randomly crop the image for the base unet'
|
||||
|
||||
# predict x0 config
|
||||
|
||||
|
||||
@@ -1 +1 @@
|
||||
__version__ = '0.24.0'
|
||||
__version__ = '0.24.1'
|
||||
|
||||
Reference in New Issue
Block a user