Views
No views yet
Qwen/Qwen2-VL-7B-Instruct using optimum-intel
via the export space.pip install optimum[openvino] transformers pillow torch1from optimum.intel import OVModelForVisualCausalLM
2from transformers import AutoProcessor
3from PIL import Image
4
5MODEL_ID = "TheAverageDetective/Qwen2-VL-7B-Instruct-openvino"
6image_path = "test.png"
7
8# Load model and processor
9model = OVModelForVisualCausalLM.from_pretrained(MODEL_ID, device="GPU")
10processor = AutoProcessor.from_pretrained(MODEL_ID)
11
12# Load image
13image = Image.open(image_path).convert("RGB")
14
15# Prepare messages
16messages = [
17 {
18 "role": "system",
19 "content": "You are a helpful assistant."
20 },
21 {
22 "role": "user",
23 "content": [
24 {"type": "image"},
25 {"type": "text", "text": "Describe this image in detail."}
26 ]
27 }
28]
29
30# Process and generate
31prompt_text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
32inputs = processor(text=[prompt_text], images=[image], return_tensors="pt")
33
34output_ids = model.generate(**inputs, max_new_tokens=150)
35result = processor.batch_decode(output_ids, skip_special_tokens=True)[0]
36
37print("\n",result.split("assistant\n")[-1].strip())
38