neonbjb/tortoise-tts
1
0
Fork 0
mirror of https://github.com/neonbjb/tortoise-tts.git synced 2026-01-24 09:30:27 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #635 from wgaylord/wgaylord-patch-1

Browse source 
Add CPU only support to hifigan_decoder.py
This commit is contained in:
manmay nakhashi 2023-10-18 23:10:58 +05:30 committed by GitHub
parent 95ca50ce27 e8c53c6a87
commit c8a3f8a3e0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tortoise/models/hifigan_decoder.py
View file

@ -230,6 +230,10 @@ class HifiganGenerator(torch.nn.Module):
if not conv_post_weight_norm:
remove_weight_norm(self.conv_post)
self.device = torch.device('cuda' if torch.cuda.is_available() else'cpu')
if torch.backends.mps.is_available():
self.device = torch.device('mps')
def forward(self, x, g=None):
"""
Args:
@ -287,7 +291,7 @@ class HifiganGenerator(torch.nn.Module):
mode="linear",
)
g = g.unsqueeze(0)
return self.forward(up_2.to("cuda"), g.transpose(1,2))
return self.forward(up_2.to(self.device), g.transpose(1,2))
def remove_weight_norm(self):
print("Removing weight norm...")