Spaces:
Running
Running
app.py
CHANGED
@@ -227,7 +227,7 @@ def greet(input):
|
|
227 |
#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
|
228 |
|
229 |
# hero, non-hero, food, spell, side-facing
|
230 |
-
|
231 |
mtx_2d = np.ones(shape) * one_hot_enc
|
232 |
ctx = mtx_2d.to(device=device).float()
|
233 |
|
|
|
227 |
#ctx = F.one_hot(torch.randint(0, 5, (32,)), 5).to(device=device).float()
|
228 |
|
229 |
# hero, non-hero, food, spell, side-facing
|
230 |
+
shape = (32, 5)
|
231 |
mtx_2d = np.ones(shape) * one_hot_enc
|
232 |
ctx = mtx_2d.to(device=device).float()
|
233 |
|