Nofated095
/
so-vits-svc
mirror of https://github.com/svc-develop-team/so-vits-svc.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update preprocess_hubert_f0.py

This commit is contained in:
Stardust·减 2023-07-22 21:35:52 +08:00 committed by GitHub
parent 61c78b4560
commit 7423deb26b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
preprocess_hubert_f0.py
View File

@ -27,7 +27,7 @@ hop_length = hps.data.hop_length
speech_encoder = hps["model"]["speech_encoder"]
def process_one(filename, hmodel,f0p,diff=False,mel_extractor=None,rank):
def process_one(filename, hmodel,f0p,diff=False,rank,mel_extractor=None):
# print(filename)
wav, sr = librosa.load(filename, sr=sampling_rate)
audio_norm = torch.FloatTensor(wav)
@ -114,7 +114,7 @@ def process_batch(file_chunk, f0p, diff=False, mel_extractor=None):
hmodel = utils.get_speech_encoder(speech_encoder, device=device)
print("Loaded speech encoder.")
for filename in tqdm(file_chunk):
process_one(filename, hmodel, f0p, diff, mel_extractor, rank)
process_one(filename, hmodel, f0p, diff, rank, mel_extractor)
def parallel_process(filenames, num_processes, f0p, diff, mel_extractor):
with ProcessPoolExecutor(max_workers=num_processes) as executor: