mrfakename commited on
Commit
fb41309
1 Parent(s): fededd1

Sync from GitHub repo

Browse files

This Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there

Files changed (1) hide show
  1. src/f5_tts/model/trainer.py +1 -1
src/f5_tts/model/trainer.py CHANGED
@@ -190,7 +190,7 @@ class Trainer:
190
  from f5_tts.infer.utils_infer import cfg_strength, load_vocoder, nfe_step, sway_sampling_coef
191
 
192
  vocoder = load_vocoder(vocoder_name=self.vocoder_name)
193
- target_sample_rate = self.accelerator.unwrap_model(self.model).mel_spec.mel_stft.sample_rate
194
  log_samples_path = f"{self.checkpoint_path}/samples"
195
  os.makedirs(log_samples_path, exist_ok=True)
196
 
 
190
  from f5_tts.infer.utils_infer import cfg_strength, load_vocoder, nfe_step, sway_sampling_coef
191
 
192
  vocoder = load_vocoder(vocoder_name=self.vocoder_name)
193
+ target_sample_rate = self.accelerator.unwrap_model(self.model).mel_spec.target_sample_rate
194
  log_samples_path = f"{self.checkpoint_path}/samples"
195
  os.makedirs(log_samples_path, exist_ok=True)
196