[Release] cpu support

demo_img.py

@@ -19,8 +19,8 @@ def img2vid(model_type, img0, img1, frame_ratio, iters):
     model.to(device)
     ckpt_path = hf_hub_download(repo_id='lalala125/AMT', filename=f'{model_type.lower()}.pth')
     print(model_type)
-    ckpt = torch.load(ckpt_path)
-    model.load_state_dict(ckpt['state_dict'], map_location=torch.device('cpu'))
+    ckpt = torch.load(ckpt_path, map_location=torch.device('cpu'))
+    model.load_state_dict(ckpt['state_dict'])
     model.eval()
     img0_t = img2tensor(img0).to(device)
     img1_t = img2tensor(img1).to(device)