Update data_collator.py

Sreyan88 · Jul 3, 2024 · 701851d · 701851d
1 parent 66f63ec
commit 701851d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hf-dev-train/transformers-main/src/transformers/data/data_collator.py b/hf-dev-train/transformers-main/src/transformers/data/data_collator.py
@@ -565,7 +565,7 @@ def load_audio(self, filename):
 
             if sr != 16000:
                 waveform = torchaudio.functional.resample(waveform, sr, 16000)
-
+                sr = 16000
             # if waveform.shape[0] > 1:
             #     waveform = waveform[0]
             #     waveform = waveform.unsqueeze(0)