Views
No views yet
1model_config = {
2 'attention_resolutions': '32, 16, 8',
3 'class_cond': False,
4 'image_size': 256,
5 'learn_sigma': True,
6 'rescale_timesteps': True,
7 'noise_schedule': 'linear',
8 'num_channels': 128,
9 'num_heads': 4,
10 'num_res_blocks': 2,
11 'resblock_updown': True,
12 'use_checkpoint': True,
13 'use_fp16': True,
14 'use_scale_shift_norm': True,
15}
1model_config = {
2 'attention_resolutions': '16',
3 'class_cond': False,
4 'diffusion_steps': 1000,
5 'rescale_timesteps': True,
6 'timestep_respacing': 'ddim100',
7 'image_size': 256,
8 'learn_sigma': True,
9 'noise_schedule': 'linear',
10 'num_channels': 128,
11 'num_heads': 1,
12 'num_res_blocks': 2,
13 'use_checkpoint': True,
14 'use_scale_shift_norm': False
15}