From 3b7e87831a4b9c3af8770983d5bb20447707aa22 Mon Sep 17 00:00:00 2001 From: Alcoft <101468238+alcoftTAO@users.noreply.github.com> Date: Sun, 13 Jul 2025 15:32:24 +0000 Subject: [PATCH] Fix bug with Intel GPUs. (#47) * Fix bug with Intel GPUs. * Updated .gitignore --- rvc/lib/infer_pack/models.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/rvc/lib/infer_pack/models.py b/rvc/lib/infer_pack/models.py index 2699bd8..7fb3d82 100644 --- a/rvc/lib/infer_pack/models.py +++ b/rvc/lib/infer_pack/models.py @@ -392,6 +392,12 @@ class SineGen(torch.nn.Module): output uv: tensor(batchsize=1, length, 1) """ with torch.no_grad(): + device = next(self.parameters(), None) + if device is not None: + device = device.device + else: + device = f0.device + align_corners = device.type != "xpu" f0 = f0[:, None].transpose(1, 2) f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim, device=f0.device) # fundamental component @@ -412,7 +418,7 @@ class SineGen(torch.nn.Module): tmp_over_one.transpose(2, 1), scale_factor=float(upp), mode="linear", - align_corners=True, + align_corners=align_corners, ).transpose(2, 1) rad_values = F.interpolate( rad_values.transpose(2, 1), scale_factor=float(upp), mode="nearest"