gitea-mirror/F5-TTS
1
0
Fork 0
mirror of https://github.com/SWivid/F5-TTS.git synced 2025-12-12 07:40:43 -08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #334 from lpscr/main

Browse Source 
small fix stuff
This commit is contained in:
Yushen CHEN
2024-10-31 11:44:58 +08:00
committed by GitHub
parent 7dd3eecf2d 6cbb548f9c
commit dee0420b59
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/f5_tts/api.py
View File

@@ -15,6 +15,9 @@ from f5_tts.infer.utils_infer import (
infer_process, infer_process,
remove_silence_for_generated_wav, remove_silence_for_generated_wav,
save_spectrogram, save_spectrogram,
preprocess_ref_audio_text,
target_sample_rate,
hop_length,
) )
@@ -31,10 +34,8 @@ class F5TTS:
): ):
# Initialize parameters # Initialize parameters
self.final_wave = None self.final_wave = None
self.target_sample_rate = 24000 self.target_sample_rate = target_sample_rate
self.n_mel_channels = 100 self.hop_length = hop_length
self.hop_length = 256
self.target_rms = 0.1
self.seed = -1 self.seed = -1
# Set device # Set device
@@ -97,6 +98,9 @@ class F5TTS:
seed = random.randint(0, sys.maxsize) seed = random.randint(0, sys.maxsize)
seed_everything(seed) seed_everything(seed)
self.seed = seed self.seed = seed
ref_file, ref_text = preprocess_ref_audio_text(ref_file, ref_text, device=self.device)
wav, sr, spect = infer_process( wav, sr, spect = infer_process(
ref_file, ref_file,
ref_text, ref_text,

2
src/f5_tts/train/finetune_gradio.py
View File

@@ -1216,7 +1216,7 @@ def infer(project, file_checkpoint, exp_name, ref_text, ref_audio, gen_text, nfe
else: else:
device_test = None device_test = None
if last_checkpoint != file_checkpoint or last_device != device_test or last_ema != use_ema: if last_checkpoint != file_checkpoint or last_device != device_test or last_ema != use_ema or tts_api is None:
if last_checkpoint != file_checkpoint: if last_checkpoint != file_checkpoint:
last_checkpoint = file_checkpoint last_checkpoint = file_checkpoint