Update app.py

app.py

@@ -29,6 +29,30 @@ model, _, _ = load_model(config, ckpt_path,
                          )
 ddim_sampler = DDIMSampler(model)
 
+def sample_denoising_batch(model, noise_shape, condition, *args,
+                           sample_type="ddim", sampler=None, 
+                           ddim_steps=None, eta=None,
+                           unconditional_guidance_scale=1.0, uc=None,
+                           denoising_progress=False,
+                           **kwargs,
+                           ):
+
+    assert(sampler is not None)
+    assert(ddim_steps is not None)
+    assert(eta is not None)
+    ddim_sampler = sampler
+    samples, _ = ddim_sampler.sample(S=ddim_steps,
+                                     conditioning=condition,
+                                     batch_size=noise_shape[0],
+                                     shape=noise_shape[1:],
+                                     verbose=denoising_progress,
+                                     unconditional_guidance_scale=unconditional_guidance_scale,
+                                     unconditional_conditioning=uc,
+                                     eta=eta,
+                                     **kwargs,
+                                    )
+    return samples
+                               
 @torch.no_grad()
 def sample_text2video(model, prompt, n_samples, batch_size,
                       sample_type="ddim", sampler=None,