|
|
|
@ -2240,8 +2240,7 @@ |
|
|
|
|
"use_secondary_model = True #@param {type: 'boolean'}\n", |
|
|
|
|
"diffusion_sampling_mode = 'ddim' #@param ['plms','ddim'] \n", |
|
|
|
|
"\n", |
|
|
|
|
"timestep_respacing = '250' #@param ['25','50','100','150','250','500','1000','ddim25','ddim50', 'ddim75', 'ddim100','ddim150','ddim250','ddim500','ddim1000'] \n", |
|
|
|
|
"diffusion_steps = 1000 #@param {type: 'number'}\n", |
|
|
|
|
"\n", |
|
|
|
|
"use_checkpoint = True #@param {type: 'boolean'}\n", |
|
|
|
|
"ViTB32 = True #@param{type:\"boolean\"}\n", |
|
|
|
|
"ViTB16 = True #@param{type:\"boolean\"}\n", |
|
|
|
@ -2333,9 +2332,9 @@ |
|
|
|
|
" model_config.update({\n", |
|
|
|
|
" 'attention_resolutions': '32, 16, 8',\n", |
|
|
|
|
" 'class_cond': False,\n", |
|
|
|
|
" 'diffusion_steps': diffusion_steps,\n", |
|
|
|
|
" 'diffusion_steps': 1000, #No need to edit this, it is taken care of later.\n", |
|
|
|
|
" 'rescale_timesteps': True,\n", |
|
|
|
|
" 'timestep_respacing': timestep_respacing,\n", |
|
|
|
|
" 'timestep_respacing': 250, #No need to edit this, it is taken care of later.\n", |
|
|
|
|
" 'image_size': 512,\n", |
|
|
|
|
" 'learn_sigma': True,\n", |
|
|
|
|
" 'noise_schedule': 'linear',\n", |
|
|
|
@ -2351,9 +2350,9 @@ |
|
|
|
|
" model_config.update({\n", |
|
|
|
|
" 'attention_resolutions': '32, 16, 8',\n", |
|
|
|
|
" 'class_cond': False,\n", |
|
|
|
|
" 'diffusion_steps': diffusion_steps,\n", |
|
|
|
|
" 'diffusion_steps': 1000, #No need to edit this, it is taken care of later.\n", |
|
|
|
|
" 'rescale_timesteps': True,\n", |
|
|
|
|
" 'timestep_respacing': timestep_respacing,\n", |
|
|
|
|
" 'timestep_respacing': 250, #No need to edit this, it is taken care of later.\n", |
|
|
|
|
" 'image_size': 256,\n", |
|
|
|
|
" 'learn_sigma': True,\n", |
|
|
|
|
" 'noise_schedule': 'linear',\n", |
|
|
|
|