Spaces:
Running
on
T4
Running
on
T4
fix bug on yaml
Browse files
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("
|
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)
|