Fix sample code by incorporating device into model, image, and text variables

README.md

@@ -25,14 +25,14 @@ import torch
 from transformers import AutoImageProcessor, AutoModel, AutoTokenizer
 
 HF_MODEL_PATH = 'line-corporation/clip-japanese-base'
+device = "cuda" if torch.cuda.is_available() else "cpu"
 tokenizer = AutoTokenizer.from_pretrained(HF_MODEL_PATH, trust_remote_code=True)
 processor = AutoImageProcessor.from_pretrained(HF_MODEL_PATH, trust_remote_code=True)
-model = AutoModel.from_pretrained(HF_MODEL_PATH, trust_remote_code=True)
-device = "cuda" if torch.cuda.is_available() else "cpu"
+model = AutoModel.from_pretrained(HF_MODEL_PATH, trust_remote_code=True).to(device)
 
 image = Image.open(io.BytesIO(requests.get('https://images.pexels.com/photos/2253275/pexels-photo-2253275.jpeg?auto=compress&cs=tinysrgb&dpr=3&h=750&w=1260').content))
-image = processor(image, return_tensors="pt")
-text = tokenizer(["犬", "猫", "象"])
+image = processor(image, return_tensors="pt").to(device)
+text = tokenizer(["犬", "猫", "象"]).to(device)
 
 with torch.no_grad():
     image_features = model.get_image_features(**image)