Spaces:
Runtime error
Runtime error
da03
commited on
Commit
·
f2aec00
1
Parent(s):
eb2c07e
main.py
CHANGED
|
@@ -96,6 +96,7 @@ def prepare_model_inputs(
|
|
| 96 |
'key_events': torch.zeros(len(itos), dtype=torch.long).to(device)
|
| 97 |
}
|
| 98 |
for key in keys_down:
|
|
|
|
| 99 |
inputs['key_events'][stoi[key]] = 1
|
| 100 |
|
| 101 |
if hidden_states is not None:
|
|
|
|
| 96 |
'key_events': torch.zeros(len(itos), dtype=torch.long).to(device)
|
| 97 |
}
|
| 98 |
for key in keys_down:
|
| 99 |
+
key = key.lower()
|
| 100 |
inputs['key_events'][stoi[key]] = 1
|
| 101 |
|
| 102 |
if hidden_states is not None:
|