lengyue233 commited on
Commit
0c92d16
1 Parent(s): af81912

make hydra happy

Browse files
Files changed (1) hide show
  1. tools/vqgan/inference.py +2 -0
tools/vqgan/inference.py CHANGED
@@ -1,6 +1,7 @@
1
  from pathlib import Path
2
 
3
  import click
 
4
  import librosa
5
  import numpy as np
6
  import soundfile as sf
@@ -18,6 +19,7 @@ OmegaConf.register_new_resolver("eval", eval)
18
 
19
 
20
  def load_model(config_name, checkpoint_path, device="cuda"):
 
21
  with initialize(version_base="1.3", config_path="../../fish_speech/configs"):
22
  cfg = compose(config_name=config_name)
23
 
 
1
  from pathlib import Path
2
 
3
  import click
4
+ import hydra
5
  import librosa
6
  import numpy as np
7
  import soundfile as sf
 
19
 
20
 
21
  def load_model(config_name, checkpoint_path, device="cuda"):
22
+ hydra.core.global_hydra.GlobalHydra.instance().clear()
23
  with initialize(version_base="1.3", config_path="../../fish_speech/configs"):
24
  cfg = compose(config_name=config_name)
25