update
Browse files- README.md +1 -1
- 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"
|
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',
|