Mehdi Cherti commited on
Commit
ba46911
·
1 Parent(s): 301a4a3
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -34,7 +34,6 @@ default = "diffusion_db_128ch_1timesteps_openclip_vith14"
34
  def gen(md, model_name, md2, text, seed, nb_samples, width, height):
35
  model = load(model_name)
36
  torch.manual_seed(int(seed))
37
- model = models[model_name]
38
  nb_samples = int(nb_samples)
39
  height = int(height)
40
  width = int(width)
 
34
  def gen(md, model_name, md2, text, seed, nb_samples, width, height):
35
  model = load(model_name)
36
  torch.manual_seed(int(seed))
 
37
  nb_samples = int(nb_samples)
38
  height = int(height)
39
  width = int(width)