Views
No views yet
Pillow==10.1.0
torch==2.1.2
torchvision==0.16.2
transformers==4.40.0
sentencepiece==0.1.99
accelerate==0.30.1
bitsandbytes==0.43.11# test.py
2import torch
3from PIL import Image
4from transformers import AutoModel, AutoTokenizer
5
6model = AutoModel.from_pretrained('openbmb/MiniCPM-V-2_6-int4', trust_remote_code=True)
7tokenizer = AutoTokenizer.from_pretrained('openbmb/MiniCPM-V-2_6-int4', trust_remote_code=True)
8model.eval()
9
10image = Image.open('xx.jpg').convert('RGB')
11question = 'What is in the image?'
12msgs = [{'role': 'user', 'content': [image, question]}]
13
14res = model.chat(
15 image=None,
16 msgs=msgs,
17 tokenizer=tokenizer
18)
19print(res)
20
21## if you want to use streaming, please make sure sampling=True and stream=True
22## the model.chat will return a generator
23res = model.chat(
24 image=None,
25 msgs=msgs,
26 tokenizer=tokenizer,
27 sampling=True,
28 temperature=0.7,
29 stream=True
30)
31
32generated_text = ""
33for new_text in res:
34 generated_text += new_text
35 print(new_text, flush=True, end='')