Spaces:
Running
Running
torch
Browse files
app.py
CHANGED
@@ -230,7 +230,7 @@ def greet(input):
|
|
230 |
one_hot_enc = np.array([1, 0, 0, 0, 0])
|
231 |
shape = (32, 5)
|
232 |
mtx_2d = np.ones(shape) * one_hot_enc
|
233 |
-
ctx =
|
234 |
|
235 |
samples, intermediate = sample_ddim_ctx(32, ctx, n=steps)
|
236 |
|
|
|
230 |
one_hot_enc = np.array([1, 0, 0, 0, 0])
|
231 |
shape = (32, 5)
|
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_ctx(32, ctx, n=steps)
|
236 |
|