ctx

app.py

@@ -223,7 +223,17 @@ def sample_ddpm(n_sample, save_rate=20):
 
 def greet(input):
 	steps = int(input)
-	samples, intermediate = sample_ddim(32, n=steps)
+
+	#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
+
+	# hero, non-hero, food, spell, side-facing
+    shape = (32, 5)
+	mtx_2d = np.ones(shape) * one_hot_enc
+	ctx = mtx_2d.to(device=device).float()
+
+	samples, intermediate = sample_ddim_ctx(32, ctx, n=steps)
+
+	#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(steps)