diff --git a/infer/modules/train/extract_feature_print.py b/infer/modules/train/extract_feature_print.py index b9eef8a..22755b5 100644 --- a/infer/modules/train/extract_feature_print.py +++ b/infer/modules/train/extract_feature_print.py @@ -21,8 +21,9 @@ i_gpu = sys.argv[4] # but callers may pass "cuda:0", "cuda:0-cuda:1", etc. Strip the prefix # and normalise separators so any combination works. import re -i_gpu = re.sub(r'cuda:', '', str(i_gpu)) -i_gpu = i_gpu.replace('-', ',') + +i_gpu = re.sub(r"cuda:", "", str(i_gpu)) +i_gpu = i_gpu.replace("-", ",") os.environ["CUDA_VISIBLE_DEVICES"] = i_gpu exp_dir = sys.argv[5] version = sys.argv[6] diff --git a/infer/modules/train/train.py b/infer/modules/train/train.py index 8c32671..c8b4875 100644 --- a/infer/modules/train/train.py +++ b/infer/modules/train/train.py @@ -127,7 +127,9 @@ def main(): children[i].join() -def run(rank, n_gpus, hps: utils.HParams, logger: logging.Logger, master_port: str = "29500"): +def run( + rank, n_gpus, hps: utils.HParams, logger: logging.Logger, master_port: str = "29500" +): global global_step if rank == 0: # logger = utils.get_logger(hps.model_dir) diff --git a/tools/web/infer-only.py b/tools/web/infer-only.py index 218788d..037031b 100644 --- a/tools/web/infer-only.py +++ b/tools/web/infer-only.py @@ -40,11 +40,9 @@ app = gr.Blocks() with app: with gr.Tabs(): with gr.TabItem("在线demo"): - gr.Markdown( - value=""" + gr.Markdown(value=""" RVC 在线demo - """ - ) + """) sid = gr.Dropdown(label=i18n("Inferencing voice"), choices=sorted(names)) with gr.Column(): spk_item = gr.Slider( diff --git a/web.py b/web.py index c493ec4..df7f11c 100644 --- a/web.py +++ b/web.py @@ -36,7 +36,6 @@ import threading import shutil import logging - logging.getLogger("numba").setLevel(logging.WARNING) logging.getLogger("httpx").setLevel(logging.WARNING)