func -> call

rvc/wrapper/cli/handler/infer.py

@@ -4,6 +4,7 @@ from pathlib import Path
 import click
 from dotenv import load_dotenv
 from scipy.io import wavfile
+import torch
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 
@@ -129,4 +130,6 @@ def infer(
         wavfile.write(outputpath, tgt_sr, audio_opt)
         click.echo(times)
         click.echo(f"Finish inference. Check {outputpath}")
-    return tgt_sr, audio_opt, times
+
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()

rvc/wrapper/cli/handler/uvr5.py

@@ -38,8 +38,5 @@ from rvc.modules.uvr5.modules import UVR
     help="output Format",
 )
 def uvr(modelname, inputpath, outputpath, format):
-    uvr_module = UVR()
-    uvr_module.uvr_wrapper(
-        inputpath, outputpath, model_name=modelname, export_format=format
-    )
+    UVR()(inputpath, outputpath, model_name=modelname, export_format=format)
     click.echo(f"Finish uvr5. Check {outputpath}")