Yushen CHEN commited on
Commit
552c0fd
·
unverified ·
1 Parent(s): 27d98a5

Update prepare_csv_wavs.py

Browse files
src/f5_tts/train/datasets/prepare_csv_wavs.py CHANGED
@@ -54,8 +54,7 @@ def prepare_csv_wavs_dir(input_dir):
54
 
55
  def get_audio_duration(audio_path):
56
  audio, sample_rate = torchaudio.load(audio_path)
57
- num_channels = audio.shape[0]
58
- return audio.shape[1] / (sample_rate * num_channels)
59
 
60
 
61
  def read_audio_text_pairs(csv_file_path):
 
54
 
55
  def get_audio_duration(audio_path):
56
  audio, sample_rate = torchaudio.load(audio_path)
57
+ return audio.shape[1] / sample_rate
 
58
 
59
 
60
  def read_audio_text_pairs(csv_file_path):