import cv2, torchfrom PIL import Imagefrom transformers import AutoProcessor, Qwen2_5_VLForConditionalGenerationfrom qwen_vl_utils import process_vision_info MODEL_PATH = "vankey/DocShield-7B" model = Qwen2_5_VLForConditionalGeneration.from_pretrained( MODEL_PATH, torch_dtype=torch.float32, attn_implementation="eager", device_map="auto",)model.eval()processor = AutoProcessor.from_pretrained(MODEL_PATH) SYSTEM_PROMPT = ( "你是一个图像鉴伪专家,擅长结合视觉,文字结合伪造特征分析手段鉴别输入图像的真假。" "分析过程中,你会逐步分析,抽丝剥茧,找到图像伪造的蛛丝马迹,最终给出专业的鉴别结果及分析。")USER_PROMPT = "请帮我分析这张图片是否是伪造的,并给出分析报告." image = cv2.cvtColor(cv2.resize(cv2.imread("image.jpg"), (1344, 896)), cv2.COLOR_BGR2RGB)image = Image.fromarray(image) messages = [ {"role": "system", "content": SYSTEM_PROMPT}, {"role": "user", "content": [ {"type": "image", "image": image}, {"type": "text", "text": USER_PROMPT}, ]},] text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)image_inputs, video_inputs = process_vision_info(messages)inputs = processor(text=[text], images=image_inputs, videos=video_inputs, padding=True, return_tensors="pt").to(model.device) with torch.no_grad(): out = model.generate(**inputs, do_sample=True, temperature=1.0, top_p=1.0, top_k=0, repetition_penalty=1.0, max_new_tokens=8192) generated = out[:, inputs["input_ids"].shape[1]:]print(processor.batch_decode(generated, skip_special_tokens=True)[0])