make it work on cpu

tortoise/models/hifigan_decoder.py

@@ -229,7 +229,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 +290,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...")