RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

[HWH-2000]

tiny error in demo.ipynb Image-Text Matching error: image = load_demo_image(image_size=image_size,device=device) model = model.to(device='cpu')

right code: image = load_demo_image(image_size=image_size,device=device) model = model.to(device)