Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2hf_path = 'jiajunlong/TinyLLaVA-OpenELM-450M-SigLIP-0.89B'
3model = AutoModelForCausalLM.from_pretrained(hf_path, trust_remote_code=True)
4model.cuda()
5config = model.config
6tokenizer = AutoTokenizer.from_pretrained(hf_path, use_fast=False, model_max_length = config.tokenizer_model_max_length,padding_side = config.tokenizer_padding_side)
7prompt="What are these?"
8image_url="http://images.cocodataset.org/test-stuff2017/000000000001.jpg"
9output_text, genertaion_time = model.chat(prompt=prompt, image=image_url, tokenizer=tokenizer)
10print('model output:', output_text)
11print('runing time:', genertaion_time)| model_name | gqa | textvqa | sqa | vqav2 | MME | MMB | MM-VET |
|---|---|---|---|---|---|---|---|
| TinyLLaVA-1.5B | 60.3 | 51.7 | 60.3 | 76.9 | 1276.5 | 55.2 | 25.8 |
| TinyLLaVA-0.89B | 53.87 | 44.02 | 54.09 | 71.74 | 1118.75 | 37.8 | 20 |