fumiama/Retrieval-based-Voice-Conversion-WebUI
1
0
Fork 0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-05 01:10:22 +08:00
Code Activity

optimize(infer): move ipex into rvc

Browse Source
This commit is contained in:
源文雨
2024-06-14 22:01:39 +09:00
parent cc11ad460d
commit e936e24a91
8 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
infer/modules/train/train.py
View File

@@ -24,8 +24,7 @@ try:
import intel_extension_for_pytorch as ipex # pylint: disable=import-error, unused-import
if torch.xpu.is_available():
from infer.modules.ipex import ipex_init
from infer.modules.ipex.gradscaler import gradscaler_init
from rvc.ipex import ipex_init, gradscaler_init
from torch.xpu.amp import autocast
GradScaler = gradscaler_init()