BUAADreamer
/

Yi-VL-6B-hf

Visual Question Answering

Inference Endpoints

Model card Files Files and versions Community

BUAADreamer commited on 19 days ago

Commit

b8a909b

•

1 Parent(s): 0e71cb1

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -65,7 +65,7 @@ processor = AutoProcessor.from_pretrained(model_id)
 text = [processor.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=False)]
 images = [Image.open(requests.get(image_file, stream=True).raw)]
-inputs = processor(text=prompt, images=images, return_tensors='pt').to(0, torch.float16)
 output = model.generate(**inputs, max_new_tokens=200)
 output = processor.batch_decode(output, skip_special_tokens=True)

 text = [processor.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=False)]
 images = [Image.open(requests.get(image_file, stream=True).raw)]
+inputs = processor(text=text, images=images, return_tensors='pt').to(0, torch.float16)
 output = model.generate(**inputs, max_new_tokens=200)
 output = processor.batch_decode(output, skip_special_tokens=True)