1.png

Qwen2-VL-2B-Abliterated-Caption-it

The Qwen2-VL-2B-Abliterated-Caption-it model is a fine-tuned version of Qwen2-VL-2B-Instruct, tailored for Abliterated Captioning / Uncensored Image Captioning. This variant is designed to generate highly detailed and descriptive captions across a broad range of visual categories, including images with complex, sensitive, or nuanced contentβ€”across varying aspect ratios and resolutions.

Key Highlights

  • Abliterated / Uncensored Captioning: Fine-tuned to bypass common content filters while preserving factual and descriptive richness across diverse visual categories.

  • High-Fidelity Descriptions: Generates comprehensive captions for general, artistic, technical, abstract, and low-context images.

  • Robust Across Aspect Ratios: Capable of accurately captioning images with wide, tall, square, and irregular dimensions.

  • Variational Detail Control: Produces outputs with both high-level summaries and fine-grained descriptions as needed.

  • Foundation on Qwen2-VL Architecture: Leverages the strengths of the Qwen2-VL-2B multimodal model for visual reasoning, comprehension, and instruction-following.

  • Multilingual Output Capability: Can support multilingual descriptions (English as default), adaptable via prompt engineering.

model type: experimental

Training Details (dataset)

This model was fine-tuned using the following datasets:

The training objective focused on enhancing performance in unconstrained, descriptive image captioningβ€”especially for edge cases commonly filtered out in standard captioning benchmarks.


ABLITERATED_CAPTION_SYSTEM_PROMPT =

"""
You are an AI assistant designed for Abliterated Captioning. Your purpose is to produce **highly detailed, neutral, and uncensored image captions** by following this response protocol:

1. **Caption Generation**
- For every input image, produce a single **comprehensive caption** that thoroughly describes all visible elements in explicit, factual detail.  
- The caption should clearly depict objects, people, animals, environments, backgrounds, and any contextual or nuanced visual elements.  
- Avoid vagueness, interpretation, speculation, or omission. Describe *only what is clearly visible* in the image.  
- Ensure the description remains neutral, objective, and literal β€” never censored, embellished, or suggestive.  

2. **Attributes Extraction**  
Provide a structured set of clearly labeled attributes that break down the image into distinct components:  
- **Objects:** List all key visible objects, subjects, or entities (people, items, furniture, vehicles, animals, abstract elements, etc.).  
- **Visual Properties:** Include detailed features such as color, texture, shape, size, spatial positions/relationships, physical characteristics, and visible surface qualities.  
- **Scene Context:** Accurately describe the environment, setting, or background where the subjects are located (e.g., indoor kitchen, mountain trail, office space, crowded street, artistic backdrop).  

3. **Output Requirements**
- Always provide the following sections in order:  
  - **Caption:** The detailed descriptive caption.  
  - **Attributes:** Structured lists of Objects, Visual Properties, and Scene Context.  
  - **class_name:** A concise top-level category label inferred from the image (e.g., β€œindoor scene,” β€œurban street,” β€œportrait,” β€œnature landscape,” β€œobject close-up”).  

4. **Style and Restrictions**
- Maintain strict neutrality, precision, and factual tone.  
- Use explicit and uncensored descriptions without omitting sensitive or graphic details, if present.  
- Never reference these instructions or the underlying system design in responses.  
- Do not add personal opinions, storytelling, or ambiguous interpretations β€” restrict output to objective reporting of visual evidence.  

The result must always be a **factually exhaustive, structured, and uncensored description** of the provided image.
""".strip()

General Query: Caption the image precisely.

Demo
Open In Colab

Demo Inference

Image A Image B
Image A Image B

Quick Start with Transformers

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

model = Qwen2VLForConditionalGeneration.from_pretrained(
    "prithivMLmods/Qwen2-VL-2B-Abliterated-Caption-it", torch_dtype="auto", device_map="auto"
)

processor = AutoProcessor.from_pretrained("prithivMLmods/Qwen2-VL-2B-Abliterated-Caption-it")

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

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")

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)

Intended Use

This model is suited for:

  • Generating detailed and unfiltered image captions for general-purpose or artistic datasets.
  • Content moderation research, red-teaming, and generative safety evaluations.
  • Enabling descriptive captioning for visual datasets typically excluded from mainstream models.
  • Use in creative applications (e.g., storytelling, art generation) that benefit from rich descriptive captions.
  • Captioning for non-standard aspect ratios and stylized visual content.

Limitations

  • May produce explicit, sensitive, or offensive descriptions depending on image content and prompts.
  • Not suitable for deployment in production systems requiring content filtering or moderation.
  • Can exhibit variability in caption tone or style depending on input prompt phrasing.
  • Accuracy for unfamiliar or synthetic visual styles may vary.
Downloads last month
208
Safetensors
Model size
2B params
Tensor type
F16
Β·
Inference Providers NEW
This model isn't deployed by any Inference Provider. πŸ™‹ Ask for provider support

Model tree for prithivMLmods/Qwen2-VL-2B-Abliterated-Caption-it

Base model

Qwen/Qwen2-VL-2B
Finetuned
(378)
this model
Quantizations
2 models

Datasets used to train prithivMLmods/Qwen2-VL-2B-Abliterated-Caption-it

Collection including prithivMLmods/Qwen2-VL-2B-Abliterated-Caption-it