L0SG commited on
Commit
89fe793
1 Parent(s): 12d6a95
Files changed (1) hide show
  1. bigvgan.py +1 -1
bigvgan.py CHANGED
@@ -324,7 +324,7 @@ class BigVGAN(
324
  print(f"[WARNING] You have specified use_cuda_kernel=True during BigVGAN.from_pretrained(). Only inference is supported (training is not implemented)!")
325
  print(f"[WARNING] You need nvcc and ninja installed in your system that matches your PyTorch build is using to build the kernel. If not, the model will fail to initialize or generate incorrect waveform!")
326
  print(f"[WARNING] For detail, see the official GitHub repository: https://github.com/NVIDIA/BigVGAN?tab=readme-ov-file#using-custom-cuda-kernel-for-synthesis")
327
- model = cls(h, use_cuda_kernel=use_cuda_kernel)
328
 
329
  ##################################################################
330
  # download and load pretrained generator weight
 
324
  print(f"[WARNING] You have specified use_cuda_kernel=True during BigVGAN.from_pretrained(). Only inference is supported (training is not implemented)!")
325
  print(f"[WARNING] You need nvcc and ninja installed in your system that matches your PyTorch build is using to build the kernel. If not, the model will fail to initialize or generate incorrect waveform!")
326
  print(f"[WARNING] For detail, see the official GitHub repository: https://github.com/NVIDIA/BigVGAN?tab=readme-ov-file#using-custom-cuda-kernel-for-synthesis")
327
+ model = cls(h, use_cuda_kernel=use_cuda_kernel)
328
 
329
  ##################################################################
330
  # download and load pretrained generator weight