improved-diffusion
improved-diffusion copied to clipboard
How to generate 256x256 or 128x128 size image please?
I can only generate 64x64 images with a 128x128 dataset, adjusting the image_size in script_util to 128 will show 'unsupported image size', adjusting it to 256 will show ' Error(s) in loading state_dict for UNetModel. Can anyone tell me what to do?
This model doesn't support 128x128 images. Instead use OpenAI Guided Diffusion which adds support for 128x128 images and also class conditioning.