andreagasparini commited on
Commit
25ca1d9
1 Parent(s): d02ba4f

Fixes typo

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -103,7 +103,7 @@ processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-large-960h-lv60
103
 
104
  def map_to_pred(batch):
105
  inputs = processor(batch["audio"]["array"], return_tensors="pt", padding="longest")
106
- input_values = inputs.input_values.to s("cuda")
107
  attention_mask = inputs.attention_mask.to("cuda")
108
 
109
  with torch.no_grad():
 
103
 
104
  def map_to_pred(batch):
105
  inputs = processor(batch["audio"]["array"], return_tensors="pt", padding="longest")
106
+ input_values = inputs.input_values.to("cuda")
107
  attention_mask = inputs.attention_mask.to("cuda")
108
 
109
  with torch.no_grad():