Logic/DDT
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

app demo

Browse Source
This commit is contained in:
wangshuai6
2025-04-11 18:00:29 +08:00
parent 12227c4f7f
commit d961ccc57a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app.py
View File

@@ -129,7 +129,7 @@ if __name__ == "__main__":
timeshift=1.0 timeshift=1.0
) )
ckpt_path = os.path.join(args.ckpt_path, "model.ckpt") 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 = load_model(ckpt, denoiser)
denoiser = denoiser.cuda() denoiser = denoiser.cuda()
vae = vae.cuda() vae = vae.cuda()