Spaces:
Sleeping
Sleeping
fix bugs.
Browse files
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 |
-
|
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)
|