Update README.md
Browse files
README.md
CHANGED
@@ -108,7 +108,8 @@ test_dataset = test_dataset.map(speech_file_to_array_fn)
|
|
108 |
inputs = processor(test_dataset["speech"], sampling_rate=16_000,
|
109 |
return_tensors="pt", padding=True)
|
110 |
with torch.no_grad():
|
111 |
-
logits = model(inputs.input_values,
|
|
|
112 |
predicted_sentences = processor.batch_decode(
|
113 |
logits=logits.numpy(),
|
114 |
num_processes=num_processes
|
|
|
108 |
inputs = processor(test_dataset["speech"], sampling_rate=16_000,
|
109 |
return_tensors="pt", padding=True)
|
110 |
with torch.no_grad():
|
111 |
+
logits = model(inputs.input_values,
|
112 |
+
attention_mask=inputs.attention_mask).logits
|
113 |
predicted_sentences = processor.batch_decode(
|
114 |
logits=logits.numpy(),
|
115 |
num_processes=num_processes
|