Downloads · 30 days
12
3% of all-time downloads
princepride/MiniCPM-V-2_6-VPM
MiniCPM-V-2_6-VPM is a feature extraction model from princepride. Use it when you need embeddings to search or compare text. It is set up for transformers.
from transformers import AutoProcessor, AutoModel import torch from PIL import Image
Downloads · 30 days
12
3% of all-time downloads
All-time downloads
404
Public
Parameters
418M
836 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors836 MB · 100%
From the Hugging Face model README
from transformers import AutoProcessor, AutoModel
import torch
from PIL import Image
model = AutoModel.from_pretrained('princepride/MiniCPM-V-2_6-VPM', trust_remote_code=True,
attn_implementation='flash_attention_2', torch_dtype=torch.bfloat16)
processor = AutoProcessor.from_pretrained('princepride/MiniCPM-V-2_6-VPM', trust_remote_code=True)
image = Image.open(r'workspace/00002-2654981627.png').convert('RGB')
inputs = processor(
[image],
max_slice_nums=max_slice_nums,
use_image_id=use_image_id,
return_tensors="pt",
max_length=max_inp_length
)
model(inputs)
model = model.eval().cuda()