| license: cc-by-nc-4.0 | |
| #@markdown ####**Custom Model Settings:** | |
| if diffusion_model == 'custom': | |
| model_config.update({ | |
| 'attention_resolutions': '32, 16, 8', | |
| 'class_cond': False, | |
| 'diffusion_steps': 1000, | |
| 'rescale_timesteps': True, | |
| 'image_size': 512, | |
| '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, | |
| }) |