Skip to content

Commit 0fcc293

Browse files
authored
Add files via upload
1 parent 954ce83 commit 0fcc293

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

lib/infer_pack/models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,8 @@ def _f02uv(self, f0):
315315
# generate uv signal
316316
uv = torch.ones_like(f0)
317317
uv = uv * (f0 > self.voiced_threshold)
318+
if(uv.device.type=="privateuseone"):#for DirectML
319+
uv=uv.float()
318320
return uv
319321

320322
def forward(self, f0, upp):

0 commit comments

Comments
 (0)