b

app.py

@@ -222,7 +222,7 @@ def sample_ddpm(n_sample, context, save_rate=20):
     return samples, intermediate
 
 @torch.no_grad()
-def sample_ddpm_context(n_sample, save_rate=20):
+def sample_ddpm_context(n_sample, context, save_rate=20):
     # x_T ~ N(0, 1), sample initial noise
     samples = torch.randn(n_sample, 3, height, height).to(device)