jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add logging for arguments of /inference endpoint

Browse Source
This commit is contained in:
Andrey Danilov
2024-05-17 16:58:04 +03:00
parent e4cb45016f
commit 7c815afff4
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
rvc/wrapper/api/endpoints/inference.py
View File

@@ -12,6 +12,10 @@ from rvc.modules.vc.modules import VC
import glob import glob
import os import os
import logging
logger: logging.Logger = logging.getLogger(__name__)
router = APIRouter() router = APIRouter()
from dotenv import load_dotenv from dotenv import load_dotenv
@@ -44,6 +48,8 @@ def inference(
protect: float = Form(0.33), protect: float = Form(0.33),
): ):
print(res_type) print(res_type)
logger.info(f"Arguments: {locals()}")
vc = VC() vc = VC()
vc.get_vc(modelpath) vc.get_vc(modelpath)
tgt_sr, audio_opt, times, _ = vc.vc_inference( tgt_sr, audio_opt, times, _ = vc.vc_inference(