Update README.md
Browse files
README.md
CHANGED
@@ -39,10 +39,8 @@ test_ds = test_ds.map(map_to_array)
|
|
39 |
def map_to_pred(batch):
|
40 |
inputs = processor(batch["speech"], sampling_rate=16000, return_tensors="pt", padding="longest")
|
41 |
input_values = inputs.input_values.to("cuda")
|
42 |
-
#attention_mask = inputs.attention_mask.to("cuda")
|
43 |
|
44 |
with torch.no_grad():
|
45 |
-
#logits = model(input_values, attention_mask=attention_mask).logits
|
46 |
logits = model(input_values).logits
|
47 |
|
48 |
predicted_ids = torch.argmax(logits, dim=-1)
|
|
|
39 |
def map_to_pred(batch):
|
40 |
inputs = processor(batch["speech"], sampling_rate=16000, return_tensors="pt", padding="longest")
|
41 |
input_values = inputs.input_values.to("cuda")
|
|
|
42 |
|
43 |
with torch.no_grad():
|
|
|
44 |
logits = model(input_values).logits
|
45 |
|
46 |
predicted_ids = torch.argmax(logits, dim=-1)
|