[Release] cpu support

demo_vid.py

@@ -19,8 +19,8 @@ def vid2vid(model_type, video, iters):
     model = model_dict[model_type]()
     model.to(device)
     ckpt_path = hf_hub_download(repo_id='lalala125/AMT', filename=f'{model_type.lower()}.pth')
-    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()
     vcap = cv2.VideoCapture(video)
     ori_frame_rate = vcap.get(cv2.CAP_PROP_FPS)