commit 2daf2af632c55fbe5647713194a57cb57848cbbc Author: 温韵清 Date: Wed Aug 6 14:31:46 2025 +0800 feature: add test code for MiniCPM_V_2 diff --git a/MiniCPM_V_2_6/demo.jpeg b/MiniCPM_V_2_6/demo.jpeg new file mode 100644 index 0000000..9fdc040 Binary files /dev/null and b/MiniCPM_V_2_6/demo.jpeg differ diff --git a/MiniCPM_V_2_6/test.py b/MiniCPM_V_2_6/test.py new file mode 100644 index 0000000..26ea782 --- /dev/null +++ b/MiniCPM_V_2_6/test.py @@ -0,0 +1,20 @@ +import torch +from PIL import Image +from modelscope import AutoModel, AutoTokenizer +model_path = '/share/fshare/common/models/OpenBMB/MiniCPM-V-2_6' +model = AutoModel.from_pretrained(model_path, trust_remote_code=True).to(dtype=torch.bfloat16) +model = model.to(device='cuda:3') +model = model.eval() +tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True) + + +image = Image.open('demo.jpeg').convert('RGB') +question = '图片里有什么?' +msgs = [{'role': 'user', 'content': [image, question]}] + +res = model.chat( + image=None, + msgs=msgs, + tokenizer=tokenizer +) +print(res)