Commit
·
8ceaa8e
1
Parent(s):
f83abf2
aaand more tab fixes
Browse files
README.md
CHANGED
|
@@ -59,7 +59,7 @@ test_dataset = test_dataset.map(speech_file_to_array_fn)
|
|
| 59 |
inputs = processor(test_dataset["speech"][:2], sampling_rate=16_000, return_tensors="pt", padding=True)
|
| 60 |
|
| 61 |
with torch.no_grad():
|
| 62 |
-
|
| 63 |
|
| 64 |
predicted_ids = torch.argmax(logits, dim=-1)
|
| 65 |
|
|
|
|
| 59 |
inputs = processor(test_dataset["speech"][:2], sampling_rate=16_000, return_tensors="pt", padding=True)
|
| 60 |
|
| 61 |
with torch.no_grad():
|
| 62 |
+
logits = model(inputs.input_values, attention_mask=inputs.attention_mask).logits
|
| 63 |
|
| 64 |
predicted_ids = torch.argmax(logits, dim=-1)
|
| 65 |
|