jzq11111 commited on
Commit
19b7021
·
verified ·
1 Parent(s): 6c0e7a4

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