Update prepare_csv_wavs.py

lpscr · Nov 1, 2024 · 552c0fd · 552c0fd
1 parent 27d98a5
commit 552c0fd
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/f5_tts/train/datasets/prepare_csv_wavs.py b/src/f5_tts/train/datasets/prepare_csv_wavs.py
@@ -54,8 +54,7 @@ def prepare_csv_wavs_dir(input_dir):
 
 def get_audio_duration(audio_path):
     audio, sample_rate = torchaudio.load(audio_path)
-    num_channels = audio.shape[0]
-    return audio.shape[1] / (sample_rate * num_channels)
+    return audio.shape[1] / sample_rate
 
 
 def read_audio_text_pairs(csv_file_path):