From 90fc92ee9caa39c0daccfe62089c6b548b3f42ed Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Mon, 8 Apr 2024 16:14:00 +0800 Subject: [PATCH] debug --- scripts/inference-vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/inference-vae.py b/scripts/inference-vae.py index afbc06d..bc9f597 100644 --- a/scripts/inference-vae.py +++ b/scripts/inference-vae.py @@ -145,7 +145,7 @@ def main(): model=vae, dataloader=dataloader ) # load model using booster - booster.load_model(vae, os.path.join(cfg.ckpt_path, "model")) + booster.load_model(vae, os.path.join(cfg.model["from_pretrained"], "model")) # 4.1. batch generation