import torchfrom PIL import Imagefrom transformers import AutoProcessor, Qwen3_5ForConditionalGeneration model_id = "pianzhikuang/VisAlloc_Final"processor = AutoProcessor.from_pretrained(model_id)model = Qwen3_5ForConditionalGeneration.from_pretrained( model_id, dtype=torch.bfloat16, device_map="auto").eval() messages = [{"role": "user", "content": [ {"type": "image", "image": Image.open("image.jpg").convert("RGB")}, {"type": "text", "text": "Describe this image."},]}]inputs = processor.apply_chat_template( messages, tokenize=True, add_generation_prompt=True, return_dict=True, return_tensors="pt",).to(model.device)with torch.inference_mode(): output = model.generate(**inputs, max_new_tokens=256, do_sample=False)print(processor.decode( output[0, inputs["input_ids"].shape[1]:], skip_special_tokens=True))