CoolFace
Modelpublic

foolen/qwen2.5-vl-7b-cognition-full-sft

sourceHugging Faceapache-2.0updated 10mo agoView on Hugging Face
0likes10downloads
Model Card

license: apache-2.0 language:

  • —en library_name: transformers tags:
  • —vision-language-model
  • —qwen
  • —instruct
  • —sft
  • —cognition
  • —multimodal datasets:
  • —foolen/CogIP-Bench task_categories:
  • —text-generation
  • —image-to-text
  • —visual-question-answering prettyname: Qwen2.5-VL-7B-Cognition-Full-SFT pipelinetag: image-text-to-text modelcreator: Junlin basemodel: Qwen/Qwen2.5-VL-7B-Instruct modeltype: qwen2vl ---

Qwen2.5-VL-7B-Cognition-Full-SFT

This repo contains the fine-tuned Qwen-2.5-VL 7B Instruct weights (SFT) trained on the CogIP-Bench dataset which is the model demonstrating the effectiveness of cognition alignment by Qwen-Image. More details see GitHub repo.

<img width="2367" height="1335" alt="16b5a48e2028643c278bf962015db2fc" src="https://github.com/user-attachments/assets/db0c4cd3-f184-4b74-8055-dc8c0244ef5c" />

Figure: Qualitative comparison of images generated by the Qwen-Image pipeline using different LLM backbones (with the same prompt). The figure shows the effect of pretraining versus supervised fine-tuning (SFT) on image cognition properties. For each image pair, Left: Base model; right: SFT model. Generation prompts are shown under each image pair. We can see that images generated with our SFT MLLM backbone better demonstrate the cognitive cues embedded in the prompts.

Quick load example

python
from transformers import AutoTokenizer, AutoModelForCausalLM, AutoFeatureExtractor

MODEL_ID = "foolen/qwen2.5-vl-7b-cognition-full-sft"
tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
feature_extractor = AutoFeatureExtractor.from_pretrained(MODEL_ID)
model = AutoModelForCausalLM.from_pretrained(MODEL_ID, device_map="auto")  # or load_in_4bit=True