Merge pull request #28 from alembics/applying-removal-of-timestep-config-to-ipynb

Synchronize .ipynb with .py
pull/17/merge
Adam Letts 3 years ago committed by GitHub
commit a4b1c77110
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 11
      Disco_Diffusion.ipynb

@ -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",

Loading…
Cancel
Save