gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2026-01-01 15:50:03 -08:00
Code Issues Packages Projects Releases Wiki Activity

fix. update tgt_sr def for log_sample with new mel_spec module

Browse Source
This commit is contained in:
SWivid
2024-11-01 15:42:19 +08:00
parent 315230210d
commit 2a3deaab33
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/f5_tts/model/trainer.py
View File

@@ -190,7 +190,7 @@ class Trainer:
from f5_tts.infer.utils_infer import cfg_strength, load_vocoder, nfe_step, sway_sampling_coef
vocoder = load_vocoder(vocoder_name=self.vocoder_name)
target_sample_rate = self.accelerator.unwrap_model(self.model).mel_spec.mel_stft.sample_rate
target_sample_rate = self.accelerator.unwrap_model(self.model).mel_spec.target_sample_rate
log_samples_path = f"{self.checkpoint_path}/samples"
os.makedirs(log_samples_path, exist_ok=True)