yyk19 commited on
Commit
4c34c9a
1 Parent(s): b0e2a49
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -68,9 +68,9 @@ def process_multi_wrapper_only_show_rendered(rendered_txt_0, rendered_txt_1, ren
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)
74
 
75
  ddim_sampler = DDIMSampler(model)
76
  render_tool = Render_Text(model)
 
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)
74
 
75
  ddim_sampler = DDIMSampler(model)
76
  render_tool = Render_Text(model)