From 0fcc293dd0909fd1a264e8e044d6563e3c4e42db Mon Sep 17 00:00:00 2001 From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com> Date: Sat, 12 Aug 2023 22:58:41 +0800 Subject: [PATCH] Add files via upload --- lib/infer_pack/models.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/lib/infer_pack/models.py b/lib/infer_pack/models.py index eb73e78..fa45f6b 100644 --- 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):