Add files via upload

2023-08-12 22:58:41 +08:00 · 2023-08-12 22:58:41 +08:00 · 0fcc293dd0
parent 954ce83d04
commit 0fcc293dd0
1 changed files with 2 additions and 0 deletions
--- a/lib/infer_pack/models.py
+++ b/lib/infer_pack/models.py
@ -315,6 +315,8 @@ class SineGen(torch.nn.Module):
        # generate uv signal
        uv = torch.ones_like(f0)
        uv = uv * (f0 > self.voiced_threshold)
+        if(uv.device.type=="privateuseone"):#for DirectML
+            uv=uv.float()
        return uv

    def forward(self, f0, upp):