protect against random cropping for base unet

This commit is contained in:
Phil Wang
2022-07-14 12:49:26 -07:00
parent 5be1f57448
commit 1662bbf226
2 changed files with 2 additions and 1 deletions

View File

@@ -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

View File

@@ -1 +1 @@
__version__ = '0.24.0'
__version__ = '0.24.1'