VKP-Omni-2B

VKP-Omni-2B is an ultra-fast multimodal AI assistant developed by Vikash Kumar Pandit at IIT Patna.

Key Capabilities

• Visual identification and reasoning (Qwen2-VL architecture)
• Agentic live web search tool integration
• Fast English and Hindi logical reasoning

Inference Quickstart

import torch
from transformers import Qwen2VLForConditionalGeneration, AutoProcessor

model_id = "NandiAi/VKP-Omni-2B"
processor = AutoProcessor.from_pretrained(model_id)
model = Qwen2VLForConditionalGeneration.from_pretrained(
    model_id,
    torch_dtype=torch.bfloat16,
    device_map="auto"
)