Spaces:
Runtime error
Runtime error
Update modules/audio_detokenizer/bigvgan_wrapper.py
Browse files
modules/audio_detokenizer/bigvgan_wrapper.py
CHANGED
@@ -83,7 +83,8 @@ class BigVGANWrapper:
|
|
83 |
data = f.read()
|
84 |
json_config = json.loads(data)
|
85 |
h = AttrDict(json_config)
|
86 |
-
vocoder = BigVGAN(h, True)
|
|
|
87 |
state_dict_g = load_checkpoint(ckpt_path, "cpu")
|
88 |
vocoder.load_state_dict(state_dict_g["generator"])
|
89 |
|
|
|
83 |
data = f.read()
|
84 |
json_config = json.loads(data)
|
85 |
h = AttrDict(json_config)
|
86 |
+
# vocoder = BigVGAN(h, True)
|
87 |
+
vocoder = BigVGAN(h, False) # for huggingface demo
|
88 |
state_dict_g = load_checkpoint(ckpt_path, "cpu")
|
89 |
vocoder.load_state_dict(state_dict_g["generator"])
|
90 |
|