diff --git a/rvc/lib/ipex/__init__.py b/rvc/lib/ipex/__init__.py index 51fa4dd..e472362 100644 --- a/rvc/lib/ipex/__init__.py +++ b/rvc/lib/ipex/__init__.py @@ -135,9 +135,7 @@ def ipex_init(): torch.cuda.amp.GradScaler = torch.xpu.amp.GradScaler except Exception: try: - from .gradscaler import ( - gradscaler_init, - ) + from .gradscaler import gradscaler_init gradscaler_init() torch.cuda.amp.GradScaler = torch.xpu.amp.GradScaler diff --git a/rvc/lib/ipex/attention.py b/rvc/lib/ipex/attention.py index ca8b71e..af36193 100644 --- a/rvc/lib/ipex/attention.py +++ b/rvc/lib/ipex/attention.py @@ -1,7 +1,6 @@ import intel_extension_for_pytorch as ipex import torch - original_torch_bmm = torch.bmm diff --git a/rvc/lib/ipex/gradscaler.py b/rvc/lib/ipex/gradscaler.py index 3872f37..4122264 100644 --- a/rvc/lib/ipex/gradscaler.py +++ b/rvc/lib/ipex/gradscaler.py @@ -4,7 +4,6 @@ import intel_extension_for_pytorch as ipex import intel_extension_for_pytorch._C as core import torch - OptState = ipex.cpu.autocast._grad_scaler.OptState _MultiDeviceReplicator = ipex.cpu.autocast._grad_scaler._MultiDeviceReplicator _refresh_per_optimizer_state = (