Revert "func -> call"

This reverts commit 043fa4d750.
This commit is contained in:
Ftps
2024-06-02 02:34:38 +09:00
parent 790b504ba2
commit e02e00753d
6 changed files with 18 additions and 50 deletions
+1 -4
View File
@@ -4,7 +4,6 @@ from pathlib import Path
import click
from dotenv import load_dotenv
from scipy.io import wavfile
import torch
logging.getLogger("numba").setLevel(logging.WARNING)
@@ -130,6 +129,4 @@ def infer(
wavfile.write(outputpath, tgt_sr, audio_opt)
click.echo(times)
click.echo(f"Finish inference. Check {outputpath}")
if torch.cuda.is_available():
torch.cuda.empty_cache()
return tgt_sr, audio_opt, times
+4 -1
View File
@@ -38,5 +38,8 @@ from rvc.modules.uvr5.modules import UVR
help="output Format",
)
def uvr(modelname, inputpath, outputpath, format):
UVR()(inputpath, outputpath, model_name=modelname, export_format=format)
uvr_module = UVR()
uvr_module.uvr_wrapper(
inputpath, outputpath, model_name=modelname, export_format=format
)
click.echo(f"Finish uvr5. Check {outputpath}")