diff --git a/infer-web.py b/infer-web.py index 597bc87..99f413d 100644 --- a/infer-web.py +++ b/infer-web.py @@ -1518,4 +1518,4 @@ with gr.Blocks(title="RVC WebUI") as app: inbrowser=not config.noautoopen, server_port=config.listen_port, quiet=True, - ) \ No newline at end of file + ) diff --git a/infer/modules/train/extract_feature_print.py b/infer/modules/train/extract_feature_print.py index f39d3eb..7b2254b 100644 --- a/infer/modules/train/extract_feature_print.py +++ b/infer/modules/train/extract_feature_print.py @@ -116,7 +116,9 @@ else: feats = readwave(wav_path, normalize=saved_cfg.task.normalize) padding_mask = torch.BoolTensor(feats.shape).fill_(False) inputs = { - "source": feats.half().to(device) if is_half and device not in ["mps", "cpu"] else feats.to(device), + "source": feats.half().to(device) + if is_half and device not in ["mps", "cpu"] + else feats.to(device), "padding_mask": padding_mask.to(device), "output_layer": 9 if version == "v1" else 12, # layer 9 }