L0SG commited on
Commit
e23e1de
1 Parent(s): aa76dec
Files changed (1) hide show
  1. bigvgan.py +2 -1
bigvgan.py CHANGED
@@ -322,7 +322,8 @@ class BigVGAN(
322
  ##################################################################
323
  if use_cuda_kernel:
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 to build the kernel. For detail, see the official GitHub repository: https://github.com/NVIDIA/BigVGAN?tab=readme-ov-file#using-custom-cuda-kernel-for-synthesis")
 
326
  model = cls(h, use_cuda_kernel=use_cuda_kernel)
327
 
328
  ##################################################################
 
322
  ##################################################################
323
  if use_cuda_kernel:
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
  ##################################################################