Update app.py

app.py

@@ -71,14 +71,18 @@ def run_all(prompt, steps, n_images, weight, clip_guided):
     torch.manual_seed(seed)
     x = torch.randn([n_images, 3, side_y, side_x], device='cuda')
     t = torch.linspace(1, 0, steps + 1, device='cuda')[:-1]
-    step_list = utils.get_spliced_ddpm_cosine_schedule(t)
+    #step_list = utils.get_spliced_ddpm_cosine_schedule(t)
+    if model.min_t == 0:
+        step_list = utils.get_spliced_ddpm_cosine_schedule(t)
+    else:
+        step_list = utils.get_ddpm_schedule(t)
     if(not clip_guided):
         outs = sampling.plms_sample(cfg_model_fn, x, step_list, {})#, callback=display_callback)
     else:
         extra_args = {'clip_embed': clip_embed}
         cond_fn_ = cond_fn
         model_fn = make_cond_model_fn(model, cond_fn_)
-        outs = sampling.plms_sample(model_fn, x, steps, extra_args)
+        outs = sampling.plms_sample(model_fn, x, step_list, extra_args)
     images_out = []
     for i, out in enumerate(outs):
         images_out.append(utils.to_pil_image(out))