Hugging Face's logo

Spaces:
tencent
/
SongGeneration
Running on L40S

App Files Community
1
hainazhu commited on
Commit
f6b176e
·
1 Parent(s): c4aaa82

fix environment bug

Browse files
Files changed (3) hide show
  1. .gitignore +2 -1
  2. Dockerfile +1 -1
  3. levo_inference.py +1 -0
.gitignore CHANGED
@@ -4,4 +4,5 @@ sample/generated/
4
  .bash_history
5
  .config
6
  ckpt
7
- third_party
 
 
4
  .bash_history
5
  .config
6
  ckpt
7
+ third_party
8
+ .cache
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM juhayna/song-generation-levo:v0.1
2
 
3
  USER root
4
 
 
1
+ FROM juhayna/song-generation-levo:v0.2
2
 
3
  USER root
4
 
levo_inference.py CHANGED
@@ -24,6 +24,7 @@ class LeVoInference(torch.nn.Module):
24
  torch.backends.cudnn.enabled = False
25
  OmegaConf.register_new_resolver("eval", lambda x: eval(x))
26
  OmegaConf.register_new_resolver("concat", lambda *x: [xxx for xx in x for xxx in xx])
 
27
  OmegaConf.register_new_resolver("load_yaml", lambda x: list(OmegaConf.load(x)))
28
 
29
  self.cfg = OmegaConf.load(cfg_path)
 
24
  torch.backends.cudnn.enabled = False
25
  OmegaConf.register_new_resolver("eval", lambda x: eval(x))
26
  OmegaConf.register_new_resolver("concat", lambda *x: [xxx for xx in x for xxx in xx])
27
+ OmegaConf.register_new_resolver("get_fname", lambda: 'default')
28
  OmegaConf.register_new_resolver("load_yaml", lambda x: list(OmegaConf.load(x)))
29
 
30
  self.cfg = OmegaConf.load(cfg_path)