From eda578a6327a89a5b88c77be6223ec1289dffd8c Mon Sep 17 00:00:00 2001 From: Max Date: Fri, 1 Apr 2022 15:22:24 +1100 Subject: [PATCH] Removal of confusing, redundant step params. --- disco.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/disco.py b/disco.py index 44b4094..9d397fd 100644 --- a/disco.py +++ b/disco.py @@ -2132,8 +2132,7 @@ diffusion_model = "512x512_diffusion_uncond_finetune_008100" #@param ["256x256_d use_secondary_model = True #@param {type: 'boolean'} diffusion_sampling_mode = 'ddim' #@param ['plms','ddim'] -timestep_respacing = '250' #@param ['25','50','100','150','250','500','1000','ddim25','ddim50', 'ddim75', 'ddim100','ddim150','ddim250','ddim500','ddim1000'] -diffusion_steps = 1000 #@param {type: 'number'} + use_checkpoint = True #@param {type: 'boolean'} ViTB32 = True #@param{type:"boolean"} ViTB16 = True #@param{type:"boolean"} @@ -2225,9 +2224,9 @@ if diffusion_model == '512x512_diffusion_uncond_finetune_008100': model_config.update({ 'attention_resolutions': '32, 16, 8', 'class_cond': False, - 'diffusion_steps': diffusion_steps, + 'diffusion_steps': 1000, #No need to edit this, it is taken care of later. 'rescale_timesteps': True, - 'timestep_respacing': timestep_respacing, + 'timestep_respacing': 250, #No need to edit this, it is taken care of later. 'image_size': 512, 'learn_sigma': True, 'noise_schedule': 'linear', @@ -2243,9 +2242,9 @@ elif diffusion_model == '256x256_diffusion_uncond': model_config.update({ 'attention_resolutions': '32, 16, 8', 'class_cond': False, - 'diffusion_steps': diffusion_steps, + 'diffusion_steps': 1000, #No need to edit this, it is taken care of later. 'rescale_timesteps': True, - 'timestep_respacing': timestep_respacing, + 'timestep_respacing': 250, #No need to edit this, it is taken care of later. 'image_size': 256, 'learn_sigma': True, 'noise_schedule': 'linear',