Removes annoying warning: missing "sampling_rate" argument
#1
by
andreagasparini
- opened
README.md
CHANGED
@@ -102,7 +102,9 @@ model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-large-960h-lv60-self")
|
|
102 |
processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-large-960h-lv60-self")
|
103 |
|
104 |
def map_to_pred(batch):
|
105 |
-
|
|
|
|
|
106 |
input_values = inputs.input_values.to("cuda")
|
107 |
attention_mask = inputs.attention_mask.to("cuda")
|
108 |
|
|
|
102 |
processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-large-960h-lv60-self")
|
103 |
|
104 |
def map_to_pred(batch):
|
105 |
+
# LibriSpeech sampling rate (16kHz) is equal to Wav2Vec2 processor sampling rate -> pass audio directly to processor
|
106 |
+
# Note that in case the sampling rates do not match you would have to resample!
|
107 |
+
inputs = processor(batch["audio"]["array"], return_tensors="pt", padding="longest", sampling_rate=batch["audio"]["sampling_rate"])
|
108 |
input_values = inputs.input_values.to("cuda")
|
109 |
attention_mask = inputs.attention_mask.to("cuda")
|
110 |
|