L0SG commited on
Commit
d396d64
·
1 Parent(s): e23e1de
Files changed (2) hide show
  1. README.md +1 -1
  2. bigvgan.py +1 -1
README.md CHANGED
@@ -52,7 +52,7 @@ model = bigvgan.BigVGAN.from_pretrained('nvidia/bigvgan_base_22khz_80band')
52
 
53
  # remove weight norm in the model and set to eval mode
54
  model.remove_weight_norm()
55
- model.eval().to(device)
56
 
57
  import librosa
58
  from meldataset import get_mel_spectrogram
 
52
 
53
  # remove weight norm in the model and set to eval mode
54
  model.remove_weight_norm()
55
+ model = model.eval().to(device)
56
 
57
  import librosa
58
  from meldataset import get_mel_spectrogram
bigvgan.py CHANGED
@@ -333,7 +333,7 @@ class BigVGAN(
333
  print("Loading weights from local directory")
334
  model_file = os.path.join(model_id, 'bigvgan_generator.pt')
335
  else:
336
- print(f"Downloading weights from {model_id}")
337
  model_file = hf_hub_download(
338
  repo_id=model_id,
339
  filename='bigvgan_generator.pt',
 
333
  print("Loading weights from local directory")
334
  model_file = os.path.join(model_id, 'bigvgan_generator.pt')
335
  else:
336
+ print(f"Loading weights from {model_id}")
337
  model_file = hf_hub_download(
338
  repo_id=model_id,
339
  filename='bigvgan_generator.pt',