From d961ccc57a0f99a909c11241204c0b167bde04bf Mon Sep 17 00:00:00 2001 From: wangshuai6 Date: Fri, 11 Apr 2025 18:00:29 +0800 Subject: [PATCH] app demo --- app.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/app.py b/app.py index 63013c6..41b1de8 100644 --- 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()