Mantis
is a multimodal conversational AI model that can chat with users about images and text. It's optimized for multi-image reasoning, where interleaved text and images can be used to generate responses.
Note that this is an older version of Mantis
, please refer to our newest version at
mantis-Siglip-llama3
. The newer version improves significantly over both multi-image and single-image tasks.
Mantis is trained on the newly curated dataset
Mantis-Instruct
, a large-scale multi-image QA dataset that covers various multi-image reasoning tasks.
from mantis.models.mllava import chat_mllava
from PIL import Image
import torch
image1 = "image1.jpg"
image2 = "image2.jpg"
images = [Image.open(image1), Image.open(image2)]
# load processor and modelfrom mantis.models.mllava import MLlavaProcessor, LlavaForConditionalGeneration
processor = MLlavaProcessor.from_pretrained("TIGER-Lab/Mantis-bakllava-7b")
model = LlavaForConditionalGeneration.from_pretrained("TIGER-Lab/Mantis-bakllava-7b", device_map="auto", torch_dtype=torch.bfloat16, attn_implementation="flash_attention_2")
# chat
text = "<image> <image> What's the difference between these two images? Please describe as much as you can."
response, history = chat_mllava(text, images, model, processor)
print("USER: ", text)
print("ASSISTANT: ", response)
# The image on the right has a larger number of wallets displayed compared to the image on the left. The wallets in the right image are arranged in a grid pattern, while the wallets in the left image are displayed in a more scattered manner. The wallets in the right image have various colors, including red, purple, and brown, while the wallets in the left image are primarily brown.
text = "How many items are there in image 1 and image 2 respectively?"
response, history = chat_mllava(text, images, model, processor, history=history)
print("USER: ", text)
print("ASSISTANT: ", response)
# There are two items in image 1 and four items in image 2.
Or, you can run the model without relying on the mantis codes, using pure hugging face transformers. See
examples/run_mantis_hf.py
for details.
Training
Training codes will be released soon.
Runs of TIGER-Lab Mantis-llava-7b on huggingface.co
7
Total runs
0
24-hour runs
0
3-day runs
3
7-day runs
-46
30-day runs
More Information About Mantis-llava-7b huggingface.co Model
Mantis-llava-7b huggingface.co is an AI model on huggingface.co that provides Mantis-llava-7b's model effect (), which can be used instantly with this TIGER-Lab Mantis-llava-7b model. huggingface.co supports a free trial of the Mantis-llava-7b model, and also provides paid use of the Mantis-llava-7b. Support call Mantis-llava-7b model through api, including Node.js, Python, http.
Mantis-llava-7b huggingface.co is an online trial and call api platform, which integrates Mantis-llava-7b's modeling effects, including api services, and provides a free online trial of Mantis-llava-7b, you can try Mantis-llava-7b online for free by clicking the link below.
TIGER-Lab Mantis-llava-7b online free url in huggingface.co:
Mantis-llava-7b is an open source model from GitHub that offers a free installation service, and any user can find Mantis-llava-7b on GitHub to install. At the same time, huggingface.co provides the effect of Mantis-llava-7b install, users can directly use Mantis-llava-7b installed effect in huggingface.co for debugging and trial. It also supports api for free installation.