@@ -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
|
||||
|
||||
@@ -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}")
|
||||
|
||||
Reference in New Issue
Block a user