app demo

2025-04-11 18:00:29 +08:00
parent 12227c4f7f
commit d961ccc57a
1 changed files with 1 additions and 1 deletions
--- a/app.py
+++ b/app.py
@@ -129,7 +129,7 @@ if __name__ == "__main__":
       timeshift=1.0
    )
    ckpt_path = os.path.join(args.ckpt_path, "model.ckpt")
-    ckpt = torch.load(args.ckpt_path, map_location="cpu")
+    ckpt = torch.load(ckpt_path, map_location="cpu")
    denoiser = load_model(ckpt, denoiser)
    denoiser = denoiser.cuda()
    vae = vae.cuda()