diffuserfix

audioldm/bigvgan/__init__.py

@@ -3,11 +3,15 @@ import bigvgan
 from huggingface_hub import hf_hub_download
 
 class BigVGANVocoder:
-    def __init__(self, device='cuda', use_cuda_kernel=False):
+    def __init__(self, device=None):
+        # Set default device if none provided
+        if device is None:
+            device = 'cuda' if torch.cuda.is_available() else 'cpu'
+            
         # Load the pretrained model
         self.model = bigvgan.BigVGAN.from_pretrained(
             'nvidia/bigvgan_v2_44khz_128band_512x',
-            use_cuda_kernel=use_cuda_kernel
+            use_cuda_kernel=(device == 'cuda')
         )
         self.model.remove_weight_norm()
         self.model.eval().to(device)