Views
No views yet
1 import torch
2 from transformers import AutoProcessor, AutoModelForCausalLM
3 model_name = "Hon-Wong/VoRA-7B-Instuct-Anyres"
4 processor = AutoProcessor.from_pretrained(model_name, trust_remote_code=True)
5 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
6 conversation = [
7 {
8 "role":"user",
9 "content":[
10 {
11 "type":"image",
12 "url": "{image path or url}"
13 },
14 {
15 "type":"text",
16 "text":"<image> Describe this image."
17 }
18 ]
19 }
20 ]
21 model_inputs = processor.apply_chat_template(conversation, add_generation_prompt=True, tokenize=True, return_tensors='pt', return_dict=True).to(model.device)
22 gen_kwargs = {"max_new_tokens": 1024, "eos_token_id": processor.tokenizer.eos_token_id}
23
24 with torch.inference_mode():
25 outputs = model.generate(model_inputs, **gen_kwargs)
26 output_text = processor.tokenizer.batch_decode(
27 outputs, skip_special_tokens=True
28 )
29 print(output_text)