File size: 7,018 Bytes
6b24101
 
 
 
 
 
68ed2bb
6b24101
 
 
 
 
3ede923
843a215
3ede923
9cd7cb1
 
8412459
9cd7cb1
4cad0af
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9cd7cb1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8412459
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9cd7cb1
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
---
license: apache-2.0
language:
- en
base_model:
- Qwen/Qwen2-VL-2B-Instruct
pipeline_tag: image-text-to-text
library_name: transformers
tags:
- QvQ
- Qwen
- Contexr-Explainer
---
# **QvQ Step Tiny - [2B]**

*QvQ-Step-Tiny* is a step-by-step context explainer Vision-Language model based on the Qwen2-VL architecture, fine-tuned using the VCR datasets for systematic step-by-step explanations. It is built on the Qwen2VLForConditionalGeneration framework with 2.21 billion parameters and uses BF16 (Brain Floating Point 16) precision.

# **Quickstart with Transformers**

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

```python
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(
    "prithivMLmods/QvQ-Step-Tiny", torch_dtype="auto", device_map="auto"
)

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."},
        ],
    }
]

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)
```
# **Key Enhancements of QvQ-Step-Tiny**  

1. **State-of-the-Art Visual Understanding**  
   - QvQ-Step-Tiny inherits the state-of-the-art capabilities of Qwen2-VL for understanding images of various resolutions and aspect ratios.  
   - It excels on visual reasoning benchmarks such as **MathVista**, **DocVQA**, **RealWorldQA**, and **MTVQA**, making it a powerful tool for detailed visual content analysis and question answering.  

2. **Extended Video Understanding**  
   - With the ability to process and comprehend videos of over 20 minutes, QvQ-Step-Tiny supports high-quality video-based question answering, conversational dialogs, and video content generation.  
   - It ensures a systematic, step-by-step explanation of video content, which is ideal for educational, entertainment, and professional applications.  

3. **Integration with Devices and Systems**  
   - Thanks to its advanced reasoning and decision-making capabilities, QvQ-Step-Tiny can act as an intelligent agent for operating devices such as mobile phones, robots, and other automated systems.  
   - It can process visual environments alongside textual instructions to enable seamless automation and intelligent control of devices.  

4. **Multilingual Support for Text in Images**  
   - QvQ-Step-Tiny supports multilingual text recognition within images, handling English, Chinese, and a wide range of languages, including most European languages, Japanese, Korean, Arabic, and Vietnamese.  
   - This makes it an effective model for global applications, from document analysis to multi-language accessibility solutions.  

# **Intended Use**  
1. **Step-by-Step Context Explanation**: Designed to provide detailed and systematic explanations for images and videos, making it ideal for educational, analytical, and instructional tasks.  
2. **Visual Content Understanding**: Effective for analyzing visual content across diverse resolutions, aspect ratios, and formats, including documents (DocVQA) and mathematical visuals (MathVista).  
3. **Video-based Reasoning**: Supports comprehension of long-form videos (20+ minutes) for tasks like video question answering, dialog generation, and instructional content creation.  
4. **Device Integration**: Can act as an intelligent agent to automate device operations (e.g., mobile phones, robots) by understanding visual environments and processing text-based instructions.  
5. **Multilingual Visual Text Support**: Recognizes and processes multilingual text within images, making it suitable for global applications like document processing and accessibility tools.  
6. **Advanced Question Answering**: Excels in question-answering tasks involving images, videos, and multimodal data, serving as a robust tool for interactive systems.  
7. **Accessibility Enhancements**: Assists visually impaired users by explaining visual and textual content in a clear, step-by-step manner.  

# **Limitations**  
1. **Model Size Constraints**: At 2.21 billion parameters, it may not perform as well as larger models for highly complex or nuanced tasks.  
2. **Accuracy with Low-Quality Inputs**: Performance may degrade when dealing with low-resolution images, poor lighting conditions, or noisy video/audio inputs.  
3. **Specialized Training Gaps**: While strong on general benchmarks, it might struggle with niche or highly specialized domains that require additional fine-tuning.  
4. **Multilingual Text Variability**: While multilingual text recognition is supported, performance may vary across less common or highly complex languages.  
5. **Context Length Tradeoffs**: Processing very long videos (e.g., over 20 minutes) or highly dense visual data might challenge its coherence or explanation accuracy.  
6. **Device Integration Complexity**: Deploying the model for operating devices or robots may require significant engineering efforts and robust integration pipelines.  
7. **Resource-Intensive for Long Contexts**: Despite BF16 precision, tasks with extended context lengths or high-resolution inputs could demand substantial computational resources.  
8. **Ambiguity in Prompts**: Ambiguously phrased or poorly structured input prompts may lead to incomplete or inaccurate explanations.  
9. **Static Model**: The model cannot learn dynamically from user interactions or adapt its behavior without retraining.  

# **Applications**  
- **Education**: Step-by-step explanations for visual and textual content in learning materials, including images and videos.  
- **Automation**: Integrating with robotics or smart devices for performing tasks based on visual and textual data.  
- **Content Creation**: Assisting in creating or analyzing video and image-based content, such as tutorials or product demos.  
- **Accessibility**: Enhancing accessibility tools for visually impaired or multilingual users by providing clear explanations of image or video content.  
- **Global Q&A Systems**: Supporting cross-lingual question answering in images and videos for diverse user bases.