jshiffer/Retrieval-based-Voice-Conversion
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix bug with Intel GPUs. (#47)

Browse Source 
* Fix bug with Intel GPUs.

* Updated .gitignore
This commit is contained in:
Alcoft
2025-07-13 15:32:24 +00:00
committed by GitHub
parent 37f67fb607
commit 3b7e87831a
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rvc/lib/infer_pack/models.py
View File

@@ -392,6 +392,12 @@ class SineGen(torch.nn.Module):
output uv: tensor(batchsize=1, length, 1) output uv: tensor(batchsize=1, length, 1)
""" """
with torch.no_grad(): 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 = f0[:, None].transpose(1, 2)
f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim, device=f0.device) f0_buf = torch.zeros(f0.shape[0], f0.shape[1], self.dim, device=f0.device)
# fundamental component # fundamental component
@@ -412,7 +418,7 @@ class SineGen(torch.nn.Module):
tmp_over_one.transpose(2, 1), tmp_over_one.transpose(2, 1),
scale_factor=float(upp), scale_factor=float(upp),
mode="linear", mode="linear",
align_corners=True, align_corners=align_corners,
).transpose(2, 1) ).transpose(2, 1)
rad_values = F.interpolate( rad_values = F.interpolate(
rad_values.transpose(2, 1), scale_factor=float(upp), mode="nearest" rad_values.transpose(2, 1), scale_factor=float(upp), mode="nearest"