debisoft commited on
Commit
42ec853
·
1 Parent(s): ae20671
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -232,7 +232,8 @@ def greet(input):
232
  mtx_2d = np.ones(shape) * one_hot_enc
233
  ctx = torch.from_numpy(mtx_2d).to(device=device).float()
234
 
235
- samples, intermediate = sample_ddim_context(32, ctx, n=steps)
 
236
 
237
  #samples, intermediate = sample_ddim(32, n=steps)
238
  #ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
 
232
  mtx_2d = np.ones(shape) * one_hot_enc
233
  ctx = torch.from_numpy(mtx_2d).to(device=device).float()
234
 
235
+ #samples, intermediate = sample_ddim_context(32, ctx, n=steps)
236
+ samples, intermediate = sample_ddpm(steps)
237
 
238
  #samples, intermediate = sample_ddim(32, n=steps)
239
  #ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()