sample size again

#3
by radames - opened

actually if we want default to be 128

width = width or self.unet.config.sample_size * 2 ** (len(self.vae.config.block_out_channels) - 1)

16 * 2**(4- 1) = 128

Thanks. I should have tested your other change before merging. This seem to be working as expected now.

bguisard changed pull request status to merged

nice @bguisard it works! thanks

Sign up or log in to comment