Spaces:
Running
on
Zero
Running
on
Zero
mrfakename
commited on
Commit
•
fb41309
1
Parent(s):
fededd1
Sync from GitHub repo
Browse filesThis Space is synced from the GitHub repo: https://github.com/SWivid/F5-TTS. Please submit contributions to the Space there
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.
|
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 |
|