Mirrors/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-08-29 10:09:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Disable CUDA Graph for offline WebUI by default

Browse Source
This commit is contained in:
RVC-Boss
2026-07-21 14:48:28 +08:00
parent 2578b1a732
commit e0558629c9
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
webui.py
View File

@@ -1,6 +1,12 @@
import os
import shutil
# Offline WebUI keeps the CUDA Graph implementation available, but remains
# eager by default. Set RVC_OFFLINE_CUDA_GRAPH=1 to opt in for benchmarking or
# controlled deployments.
_offline_cuda_graph = os.environ.get("RVC_OFFLINE_CUDA_GRAPH", "0") == "1"
os.environ["RVC_CUDA_GRAPH"] = "1" if _offline_cuda_graph else "0"
os.environ.setdefault("OPENBLAS_NUM_THREADS", "1")
os.environ.setdefault("no_proxy", "localhost, 127.0.0.1, ::1")
os.environ.setdefault("weight_root", "assets/weights")
@@ -143,9 +149,6 @@ print(
i18n("当前设备：%s | 推理精度：%s") % (config.device, config.dtype),
flush=True,
)
cuda_graph_status = os.environ.get("RVC_CUDA_GRAPH", "0")
print(f"RVC_CUDA_GRAPH={cuda_graph_status}", flush=True)
logger.info("RVC_CUDA_GRAPH=%s", cuda_graph_status)
# GPU filtering and precision rules are shared with inference/extraction/training.
gpu_infos = list(GPU_INFOS)
gpu_indices = sorted(GPU_INDEX)