Update README.md

README.md

@@ -10,7 +10,7 @@ from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoPro
 from qwen_vl_utils import process_vision_info
 
 
-device = "cuda" if torch.cuda.is_available() else "mps"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 
 model_dir = "medieval-data/qwen2-vl-2b-catmus"
 
@@ -47,7 +47,7 @@ inputs = processor(
     padding=True,
     return_tensors="pt",
 )
-inputs = inputs.to("mps")
+inputs = inputs.to(device)
 
 # Inference: Generation of the output
 generated_ids = model.generate(**inputs, max_new_tokens=4000)