Attention

This repository provides a DepthWise-downscaled version of the Qwen/Qwen2-VL-2B-Instruct model, reducing it from 2B parameters to 1B parameters.

The model has been further pre-trained using the same template and trainer from the upcoming Qwen/QvQ-72B-Preview (which shares the same architecture as Qwen2_VL).
Stay tuned for the QvQ-17B-Preview, which will be released soon.

DepthWise Downscale Code

import copy
import torch
import json
import os
from transformers import Qwen2VLForConditionalGeneration, AutoProcessor

base_model = "Qwen/Qwen2-VL-2B-Instruct"
target_model = "/ors/tmp/Qwen2-VL-1B-DepthWise-Downscaled"

model = Qwen2VLForConditionalGeneration.from_pretrained(
    base_model,
    torch_dtype=torch.bfloat16,
    device_map="cpu",
)

min_pixels = 256 * 28 * 28
max_pixels = 1280 * 28 * 28
processor = AutoProcessor.from_pretrained(base_model, min_pixels=min_pixels, max_pixels=max_pixels)

new_model = copy.deepcopy(model)
new_processor = copy.deepcopy(processor)

optimal_part = 4

# -----------------------------------------------------------
# PARTE 1: Remover camadas de linguagem, 
# preservando a primeira e a última
# -----------------------------------------------------------
total_language_layers = len(model.model.layers)
num_layers_to_remove = total_language_layers // optimal_part

# Se o número de camadas a remover for 0, nada é removido.
if num_layers_to_remove > 0 and total_language_layers > 2:
    # Camadas originais
    original_layers = list(model.model.layers)

    # Garantir que estamos preservando a primeira e a última
    first_layer = original_layers[0]
    last_layer  = original_layers[-1]

    # Miolo = as camadas do índice 1 até o penúltimo
    middle_layers = original_layers[1:-1]

    # Definir quanto remover do início e do fim do miolo
    # para manter a remoção equilibrada
    num_remove_front = num_layers_to_remove // 2
    num_remove_back  = num_layers_to_remove - num_remove_front

    # Remover do miolo
    # Ex.: se middle_layers = 26 e precisamos remover 7,
    # -> num_remove_front = 3, num_remove_back = 4
    # Então mantemos a fatia central
    middle_layers_kept = middle_layers[num_remove_front : len(middle_layers) - num_remove_back]

    # Reconstruir a lista final de camadas
    language_layers_to_keep = [first_layer] + list(middle_layers_kept) + [last_layer]

    # Atualizar o modelo
    new_model.model.layers = torch.nn.ModuleList(language_layers_to_keep)
else:
    # Se não for possível remover (ou não faz sentido), 
    # simplesmente copia tudo
    new_model.model.layers = copy.deepcopy(model.model.layers)

# Atualizar configuração
new_model.config.num_hidden_layers = len(new_model.model.layers)

# -----------------------------------------------------------
# PARTE 2: Remover blocos visuais, 
# preservando o primeiro e o último
# -----------------------------------------------------------
total_visual_blocks = len(model.visual.blocks)
num_visual_blocks_to_remove = total_visual_blocks // optimal_part

if num_visual_blocks_to_remove > 0 and total_visual_blocks > 2:
    # Blocos originais
    original_blocks = list(model.visual.blocks)

    # Preservar o primeiro e o último
    first_block = original_blocks[0]
    last_block  = original_blocks[-1]

    # Miolo de blocos
    middle_blocks = original_blocks[1:-1]

    # Remover de forma equilibrada do meio
    num_remove_front_v = num_visual_blocks_to_remove // 2
    num_remove_back_v  = num_visual_blocks_to_remove - num_remove_front_v

    middle_blocks_kept = middle_blocks[num_remove_front_v : len(middle_blocks) - num_remove_back_v]

    # Recombinar
    visual_blocks_to_keep = [first_block] + list(middle_blocks_kept) + [last_block]

    # Atualizar o modelo
    new_model.visual.blocks = torch.nn.ModuleList(visual_blocks_to_keep)
else:
    # Se não removemos nada, mantém tudo
    new_model.visual.blocks = copy.deepcopy(model.visual.blocks)

# Atualizar configuração do visual
new_model.config.vision_config.depth = len(new_model.visual.blocks)

# -----------------------------------------------------------
# Verificações
# -----------------------------------------------------------
assert new_model.config.num_hidden_layers == len(new_model.model.layers), \
    "num_hidden_layers não corresponde ao número de camadas (linguagem)."
assert new_model.config.vision_config.depth == len(new_model.visual.blocks), \
    "depth não corresponde ao número de blocos visuais."

print(f"Config num_hidden_layers: {new_model.config.num_hidden_layers}")
print(f"Actual number of language layers: {len(new_model.model.layers)}")
print(f"Config vision_config.depth: {new_model.config.vision_config.depth}")
print(f"Actual number of visual blocks: {len(new_model.visual.blocks)}")

# -----------------------------------------------------------
# Salvar o modelo reduzido
# -----------------------------------------------------------
new_model.save_pretrained(target_model, safe_serialization=True)
new_processor.save_pretrained(target_model)

print(f"Modelo reduzido salvo em: {target_model}")

# Verificar o arquivo de configuração salvo
config_path = os.path.join(target_model, "config.json")
with open(config_path, "r") as f:
    config = json.load(f)

print(f"Saved config num_hidden_layers: {config.get('num_hidden_layers')}")
print(f"Saved vision_config.depth: {config.get('vision_config', {}).get('depth')}")

QvQ-1B-Preview

Introduction

We're excited to unveil Qwen2-VL, the latest iteration of our Qwen-VL model, representing nearly a year of innovation.

What’s New in Qwen2-VL?

Key Enhancements:

SoTA understanding of images of various resolution & ratio: Qwen2-VL achieves state-of-the-art performance on visual understanding benchmarks, including MathVista, DocVQA, RealWorldQA, MTVQA, etc.
Understanding videos of 20min+: Qwen2-VL can understand videos over 20 minutes for high-quality video-based question answering, dialog, content creation, etc.
Agent that can operate your mobiles, robots, etc.: with the abilities of complex reasoning and decision making, Qwen2-VL can be integrated with devices like mobile phones, robots, etc., for automatic operation based on visual environment and text instructions.
Multilingual Support: to serve global users, besides English and Chinese, Qwen2-VL now supports the understanding of texts in different languages inside images, including most European languages, Japanese, Korean, Arabic, Vietnamese, etc.

Model Architecture Updates:

Naive Dynamic Resolution: Unlike before, Qwen2-VL can handle arbitrary image resolutions, mapping them into a dynamic number of visual tokens, offering a more human-like visual processing experience.

Multimodal Rotary Position Embedding (M-ROPE): Decomposes positional embedding into parts to capture 1D textual, 2D visual, and 3D video positional information, enhancing its multimodal processing capabilities.

We have three models with 2, 7 and 72 billion parameters. This repo contains the instruction-tuned 7B Qwen2-VL model. For more information, visit our Blog and GitHub.

Evaluation

Image Benchmarks

Benchmark	InternVL2-8B	MiniCPM-V 2.6	GPT-4o-mini	Qwen2-VL-7B
MMMU_val	51.8	49.8	60	54.1
DocVQA_test	91.6	90.8	-	94.5
InfoVQA_test	74.8	-	-	76.5
ChartQA_test	83.3	-	-	83.0
TextVQA_val	77.4	80.1	-	84.3
OCRBench	794	852	785	845
MTVQA	-	-	-	26.3
VCR_{en easy}	-	73.88	83.60	89.70
VCR_{zh easy}	-	10.18	1.10	59.94
RealWorldQA	64.4	-	-	70.1
MME_sum	2210.3	2348.4	2003.4	2326.8
MMBench-EN_test	81.7	-	-	83.0
MMBench-CN_test	81.2	-	-	80.5
MMBench-V1.1_test	79.4	78.0	76.0	80.7
MMT-Bench_test	-	-	-	63.7
MMStar	61.5	57.5	54.8	60.7
MMVet_GPT-4-Turbo	54.2	60.0	66.9	62.0
HallBench_avg	45.2	48.1	46.1	50.6
MathVista_testmini	58.3	60.6	52.4	58.2
MathVision	-	-	-	16.3

Video Benchmarks

Benchmark	Internvl2-8B	LLaVA-OneVision-7B	MiniCPM-V 2.6	Qwen2-VL-7B
MVBench	66.4	56.7	-	67.0
PerceptionTest_test	-	57.1	-	62.3
EgoSchema_test	-	60.1	-	66.7
Video-MME_{wo/w subs}	54.0/56.9	58.2/-	60.9/63.6	63.3/69.0

Requirements

The code of Qwen2-VL has been in the latest Hugging face transformers and we advise you to build from source with command pip install git+https://github.com/huggingface/transformers, or you might encounter the following error:

KeyError: 'qwen2_vl'

Quickstart

We offer a toolkit to help you handle various types of visual input more conveniently. This includes base64, URLs, and interleaved images and videos. You can install it using the following command:

pip install qwen-vl-utils

Here we show a code snippet to show you how to use the chat model with transformers and qwen_vl_utils:

from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor
from qwen_vl_utils import process_vision_info

# default: Load the model on the available device(s)
model = Qwen2VLForConditionalGeneration.from_pretrained(
    "orion-research/QvQ-1B-Preview", torch_dtype="auto", device_map="auto"
)

# We recommend enabling flash_attention_2 for better acceleration and memory saving, especially in multi-image and video scenarios.
# model = Qwen2VLForConditionalGeneration.from_pretrained(
#     "orion-research/QvQ-1B-Preview",
#     torch_dtype=torch.bfloat16,
#     attn_implementation="flash_attention_2",
#     device_map="auto",
# )

# default processer
processor = AutoProcessor.from_pretrained("orion-research/QvQ-1B-Preview")

# The default range for the number of visual tokens per image in the model is 4-16384. You can set min_pixels and max_pixels according to your needs, such as a token count range of 256-1280, to balance speed and memory usage.
# min_pixels = 256*28*28
# max_pixels = 1280*28*28
# processor = AutoProcessor.from_pretrained("orion-research/QvQ-1B-Preview", min_pixels=min_pixels, max_pixels=max_pixels)

messages = [
    {
        "role": "user",
        "content": [
            {
                "type": "image",
                "image": "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg",
            },
            {"type": "text", "text": "Describe this image."},
        ],
    }
]

# Preparation for inference
text = processor.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
    text=[text],
    images=image_inputs,
    videos=video_inputs,
    padding=True,
    return_tensors="pt",
)
inputs = inputs.to("cuda")

# Inference: Generation of the output
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)

Without qwen_vl_utils

from PIL import Image
import requests
import torch
from torchvision import io
from typing import Dict
from transformers import Qwen2VLForConditionalGeneration, AutoTokenizer, AutoProcessor

# Load the model in half-precision on the available device(s)
model = Qwen2VLForConditionalGeneration.from_pretrained(
    "orion-research/QvQ-1B-Preview", torch_dtype="auto", device_map="auto"
)
processor = AutoProcessor.from_pretrained("orion-research/QvQ-1B-Preview")

# Image
url = "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg"
image = Image.open(requests.get(url, stream=True).raw)

conversation = [
    {
        "role": "user",
        "content": [
            {
                "type": "image",
            },
            {"type": "text", "text": "Describe this image."},
        ],
    }
]


# Preprocess the inputs
text_prompt = processor.apply_chat_template(conversation, add_generation_prompt=True)
# Excepted output: '<|im_start|>system\nYou are a helpful assistant.<|im_end|>\n<|im_start|>user\n<|vision_start|><|image_pad|><|vision_end|>Describe this image.<|im_end|>\n<|im_start|>assistant\n'

inputs = processor(
    text=[text_prompt], images=[image], padding=True, return_tensors="pt"
)
inputs = inputs.to("cuda")

# Inference: Generation of the output
output_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids = [
    output_ids[len(input_ids) :]
    for input_ids, output_ids in zip(inputs.input_ids, output_ids)
]
output_text = processor.batch_decode(
    generated_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True
)
print(output_text)

Multi image inference

# Messages containing multiple images and a text query
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "image": "file:///path/to/image1.jpg"},
            {"type": "image", "image": "file:///path/to/image2.jpg"},
            {"type": "text", "text": "Identify the similarities between these images."},
        ],
    }
]

# Preparation for inference
text = processor.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
    text=[text],
    images=image_inputs,
    videos=video_inputs,
    padding=True,
    return_tensors="pt",
)
inputs = inputs.to("cuda")

# Inference
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)

Video inference

# Messages containing a images list as a video and a text query
messages = [
    {
        "role": "user",
        "content": [
            {
                "type": "video",
                "video": [
                    "file:///path/to/frame1.jpg",
                    "file:///path/to/frame2.jpg",
                    "file:///path/to/frame3.jpg",
                    "file:///path/to/frame4.jpg",
                ],
                "fps": 1.0,
            },
            {"type": "text", "text": "Describe this video."},
        ],
    }
]
# Messages containing a video and a text query
messages = [
    {
        "role": "user",
        "content": [
            {
                "type": "video",
                "video": "file:///path/to/video1.mp4",
                "max_pixels": 360 * 420,
                "fps": 1.0,
            },
            {"type": "text", "text": "Describe this video."},
        ],
    }
]

# Preparation for inference
text = processor.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True
)
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
    text=[text],
    images=image_inputs,
    videos=video_inputs,
    padding=True,
    return_tensors="pt",
)
inputs = inputs.to("cuda")

# Inference
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_text = processor.batch_decode(
    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_text)

Batch inference

# Sample messages for batch inference
messages1 = [
    {
        "role": "user",
        "content": [
            {"type": "image", "image": "file:///path/to/image1.jpg"},
            {"type": "image", "image": "file:///path/to/image2.jpg"},
            {"type": "text", "text": "What are the common elements in these pictures?"},
        ],
    }
]
messages2 = [
    {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": "Who are you?"},
]
# Combine messages for batch processing
messages = [messages1, messages1]

# Preparation for batch inference
texts = [
    processor.apply_chat_template(msg, tokenize=False, add_generation_prompt=True)
    for msg in messages
]
image_inputs, video_inputs = process_vision_info(messages)
inputs = processor(
    text=texts,
    images=image_inputs,
    videos=video_inputs,
    padding=True,
    return_tensors="pt",
)
inputs = inputs.to("cuda")

# Batch Inference
generated_ids = model.generate(**inputs, max_new_tokens=128)
generated_ids_trimmed = [
    out_ids[len(in_ids) :] for in_ids, out_ids in zip(inputs.input_ids, generated_ids)
]
output_texts = processor.batch_decode(
    generated_ids_trimmed, skip_special_tokens=True, clean_up_tokenization_spaces=False
)
print(output_texts)

More Usage Tips

For input images, we support local files, base64, and URLs. For videos, we currently only support local files.

# You can directly insert a local file path, a URL, or a base64-encoded image into the position where you want in the text.
## Local file path
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "image": "file:///path/to/your/image.jpg"},
            {"type": "text", "text": "Describe this image."},
        ],
    }
]
## Image URL
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "image": "http://path/to/your/image.jpg"},
            {"type": "text", "text": "Describe this image."},
        ],
    }
]
## Base64 encoded image
messages = [
    {
        "role": "user",
        "content": [
            {"type": "image", "image": "data:image;base64,/9j/..."},
            {"type": "text", "text": "Describe this image."},
        ],
    }
]

Image Resolution for performance boost

The model supports a wide range of resolution inputs. By default, it uses the native resolution for input, but higher resolutions can enhance performance at the cost of more computation. Users can set the minimum and maximum number of pixels to achieve an optimal configuration for their needs, such as a token count range of 256-1280, to balance speed and memory usage.

min_pixels = 256 * 28 * 28
max_pixels = 1280 * 28 * 28
processor = AutoProcessor.from_pretrained(
    "orion-research/QvQ-1B-Preview", min_pixels=min_pixels, max_pixels=max_pixels
)

Besides, We provide two methods for fine-grained control over the image size input to the model:

Define min_pixels and max_pixels: Images will be resized to maintain their aspect ratio within the range of min_pixels and max_pixels.
Specify exact dimensions: Directly set resized_height and resized_width. These values will be rounded to the nearest multiple of 28.

# min_pixels and max_pixels
messages = [
    {
        "role": "user",
        "content": [
            {
                "type": "image",
                "image": "file:///path/to/your/image.jpg",
                "resized_height": 280,
                "resized_width": 420,
            },
            {"type": "text", "text": "Describe this image."},
        ],
    }
]
# resized_height and resized_width
messages = [
    {
        "role": "user",
        "content": [
            {
                "type": "image",
                "image": "file:///path/to/your/image.jpg",
                "min_pixels": 50176,
                "max_pixels": 50176,
            },
            {"type": "text", "text": "Describe this image."},
        ],
    }
]

Limitations

While Qwen2-VL are applicable to a wide range of visual tasks, it is equally important to understand its limitations. Here are some known restrictions:

Lack of Audio Support: The current model does not comprehend audio information within videos.
Data timeliness: Our image dataset is updated until June 2023, and information subsequent to this date may not be covered.
Constraints in Individuals and Intellectual Property (IP): The model's capacity to recognize specific individuals or IPs is limited, potentially failing to comprehensively cover all well-known personalities or brands.
Limited Capacity for Complex Instruction: When faced with intricate multi-step instructions, the model's understanding and execution capabilities require enhancement.
Insufficient Counting Accuracy: Particularly in complex scenes, the accuracy of object counting is not high, necessitating further improvements.
Weak Spatial Reasoning Skills: Especially in 3D spaces, the model's inference of object positional relationships is inadequate, making it difficult to precisely judge the relative positions of objects.

These limitations serve as ongoing directions for model optimization and improvement, and we are committed to continually enhancing the model's performance and scope of application.

Citation

If you find our work helpful, feel free to give us a cite.

@article{Qwen2VL,
  title={Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolution},
  author={Wang, Peng and Bai, Shuai and Tan, Sinan and Wang, Shijie and Fan, Zhihao and Bai, Jinze and Chen, Keqin and Liu, Xuejing and Wang, Jialin and Ge, Wenbin and Fan, Yang and Dang, Kai and Du, Mengfei and Ren, Xuancheng and Men, Rui and Liu, Dayiheng and Zhou, Chang and Zhou, Jingren and Lin, Junyang},
  journal={arXiv preprint arXiv:2409.12191},
  year={2024}
}

@article{Qwen-VL,
  title={Qwen-VL: A Versatile Vision-Language Model for Understanding, Localization, Text Reading, and Beyond},
  author={Bai, Jinze and Bai, Shuai and Yang, Shusheng and Wang, Shijie and Tan, Sinan and Wang, Peng and Lin, Junyang and Zhou, Chang and Zhou, Jingren},
  journal={arXiv preprint arXiv:2308.12966},
  year={2023}
}

orion-research
/

QvQ-1B-Preview