Spaces:

stephenhoang
/

ttsStyleTTS2

Sleeping

stephenhoang commited on 25 days ago

Commit

e2d0fbc

verified ·

1 Parent(s): 56d1e65

Update inference.py

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -280,7 +280,7 @@ class StyleTTS2(torch.nn.Module):
         #     print("[WARN] phonem_head=", phonem[:200])
         # tokens = [0] + tokens + [0]
         # tokens = torch.LongTensor(tokens).unsqueeze(0).to(device)
         phonem = phonem.replace("ˈ", "").replace("ˌ", "")
         tokens = self.cleaner(phonem)

         #     print("[WARN] phonem_head=", phonem[:200])
         # tokens = [0] + tokens + [0]
         # tokens = torch.LongTensor(tokens).unsqueeze(0).to(device)
+        device = self.get_device.device
         phonem = phonem.replace("ˈ", "").replace("ˌ", "")
         tokens = self.cleaner(phonem)