WindyMadman/so-vits-svc
1
0
Fork 0
mirror of https://github.com/svc-develop-team/so-vits-svc.git synced 2025-01-08 11:57:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

Debug rank GPU

Browse Source
This commit is contained in:
ylzz1997 2023-07-23 09:43:00 +08:00
parent 3ab6f1f174
commit 174cb333f8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
preprocess_hubert_f0.py
View File

@ -114,7 +114,7 @@ def process_batch(file_chunk, f0p, diff=False, mel_extractor=None):
hmodel = utils.get_speech_encoder(speech_encoder, device=device)
logger.info(f"Loaded speech encoder for rank {rank}")
for filename in tqdm(file_chunk):
process_one(filename, hmodel, f0p, rank, diff, mel_extractor)
process_one(filename, hmodel, f0p, gpu_id, diff, mel_extractor)
def parallel_process(filenames, num_processes, f0p, diff, mel_extractor):
with ProcessPoolExecutor(max_workers=num_processes) as executor: