dn commited on
Commit
79538bb
1 Parent(s): c97b27f

fix bug on yaml

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ def process_multi_wrapper_only_show_rendered(rendered_txt_0, rendered_txt_1, ren
67
  only_show_rendered_image=True)
68
 
69
 
70
- cfg = OmegaConf.load("config_cuda.yaml")
71
  model = load_model_from_config(cfg, "model_wo_ema.ckpt", verbose=True)
72
  # model = load_model_from_config(cfg, "model_states.pt", verbose=True)
73
  # model = load_model_from_config(cfg, "model.ckpt", verbose=True)
 
67
  only_show_rendered_image=True)
68
 
69
 
70
+ cfg = OmegaConf.load("config.yaml")
71
  model = load_model_from_config(cfg, "model_wo_ema.ckpt", verbose=True)
72
  # model = load_model_from_config(cfg, "model_states.pt", verbose=True)
73
  # model = load_model_from_config(cfg, "model.ckpt", verbose=True)