Freiburg-AI-Research commited on
Commit
9f40168
Β·
1 Parent(s): a0c720e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,7 +58,7 @@ base_timestep_respacing = '40' #@param {type:"string"}
58
  sr_timestep_respacing = 'fast27'
59
 
60
  #@title Create base model.
61
- glide_path = 'base.pt' #@param {type:"string"}
62
  import os
63
  options = model_and_diffusion_defaults()
64
  options['use_fp16'] = has_cuda
@@ -84,7 +84,7 @@ print('total base parameters', sum(x.numel() for x in model.parameters()))
84
 
85
 
86
  #@title Create upsampler model.
87
- sr_glide_path = "upsample.pt" #@param {type:"string"}
88
 
89
 
90
  options_up = model_and_diffusion_defaults_upsampler()
 
58
  sr_timestep_respacing = 'fast27'
59
 
60
  #@title Create base model.
61
+ glide_path = 'glide-ft-13x625.pt' #@param {type:"string"}
62
  import os
63
  options = model_and_diffusion_defaults()
64
  options['use_fp16'] = has_cuda
 
84
 
85
 
86
  #@title Create upsampler model.
87
+ sr_glide_path = "glide-ft-12x5000.pt" #@param {type:"string"}
88
 
89
 
90
  options_up = model_and_diffusion_defaults_upsampler()