| license: cc-by-4.0 | |
| 3750 images from esa/nasa of deep space. JWST/Hubble shots mainly. | |
| model_config.update({ | |
| 'attention_resolutions': '32, 16, 8', | |
| 'class_cond': False, | |
| 'diffusion_steps': 1000, | |
| 'rescale_timesteps': True, | |
| 'timestep_respacing': 'ddim100', | |
| 'image_size': 256, | |
| 'learn_sigma': True, | |
| 'noise_schedule': 'linear', | |
| 'num_channels': 128, | |
| 'num_heads': 4, | |
| 'num_res_blocks': 2, | |
| 'resblock_updown': True, | |
| 'use_checkpoint': use_checkpoint, | |
| 'use_fp16': True, | |
| 'use_scale_shift_norm': True, | |
| }) |