RVC/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-02-16 16:20:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 卸载音色省显存

Browse Source 
顺便将所有print换成了统一的logger
This commit is contained in:
源文雨
2023-09-01 15:18:08 +08:00
parent 8d5a77dbe9
commit 04a33b9709
23 changed files with 189 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
infer/modules/uvr5/modules.py
View File

@@ -1,5 +1,7 @@
import os
import traceback
import logging
logger = logging.getLogger(__name__)
import ffmpeg
import torch
@@ -92,5 +94,5 @@ def uvr(model_name, inp_root, save_root_vocal, paths, save_root_ins, agg, format
traceback.print_exc()
if torch.cuda.is_available():
torch.cuda.empty_cache()
print("Executed torch.cuda.empty_cache()")
logger.info("Executed torch.cuda.empty_cache()")
yield "\n".join(infos)