Views
No views yet
pip install torchao;
pip install git+https://github.com/mobiusml/gemlite.git;
pip install qwen-vl-utils[decord]==0.0.8;1from transformers import Qwen2_5_VLForConditionalGeneration, AutoTokenizer, AutoProcessor
2from qwen_vl_utils import process_vision_info
3improt torch
4
5model_id = "mobiuslabsgmbh/Qwen2.5-VL-7B-Instruct_gemlite-ao_a8w8"
6model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
7 model_id, torch_dtype=torch.float16, device_map="cuda",
8 #attn_implementation="flash_attention_2",
9)
10
11processor = AutoProcessor.from_pretrained(model_id)
12
13messages = [
14 {
15 "role": "user",
16 "content": [
17 {
18 "type": "image",
19 "image": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg",
20 },
21 {"type": "text", "text": "Describe this image."},
22 ],
23 }
24]
25
26# Preparation for inference
27text = processor.apply_chat_template(
28 messages, tokenize=False, add_generation_prompt=True
29)
30image_inputs, video_inputs = process_vision_info(messages)
31inputs = processor(
32 text=[text],
33 images=image_inputs,
34 videos=video_inputs,
35 padding=True,
36 return_tensors="pt",
37)
38inputs = inputs.to("cuda")
39
40# Inference: Generation of the output
41generated_ids = model.generate(**inputs, max_new_tokens=128)
42generated_ids_trimmed = [
43 out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
44]
45output_text = processor.batch_decode(
46 generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
47)
48print(output_text)1import torch
2from vllm import LLM
3from vllm.sampling_params import SamplingParams
4
5model_id = "mobiuslabsgmbh/Qwen2.5-VL-7B-Instruct_gemlite-ao_a8w8"
6processor_args = {
7 'limit_mm_per_prompt': {"image": 3},
8 'mm_processor_kwargs': {"min_pixels": 28 * 28, "max_pixels": 1280 * 28 * 28},
9 'disable_mm_preprocessor_cache': False,
10}
11
12llm = LLM(model=model_id, gpu_memory_utilization=0.9, dtype=torch.float16, max_model_len=4096,
13 max_num_batched_tokens=4096, **processor_args)
14
15sampling_params = SamplingParams(max_tokens=1024, temperature=0.5, repetition_penalty=1.1, ignore_eos=False)
16
17messages = [{"content": "You are a helpful assistant", "role":"system"}, {"content":"Solve this equation x^2 + 1 = -1.", "role":"user"}]
18outputs = llm.chat(messages, sampling_params, chat_template=llm.get_tokenizer().chat_template)
19print(outputs[0].outputs[0].text)
20