s

app.py

@@ -226,7 +226,7 @@ def greet(input):
 	#samples, intermediate = sample_ddim(32, n=steps)
 	#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
 	#samples, intermediate = sample_ddim_context(32, ctx, steps)
-	samples, intermediate = sample_ddpm(32, steps)
+	samples, intermediate = sample_ddpm(steps)
 	#response = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[-1]))
 	#response2 = transform2(transform(np.moveaxis(samples.detach().cpu().numpy(),1,3)[1]))
 	#response = im.fromarray(intermediate[24][0][1]).convert("RGB")