Dimple 7B

🚀 Dimple-7B

Dimple is the first Discrete Diffusion Multimodal Large Language Model (DMLLM) that combines autoregressive and diffusion-based instruction tuning in a hybrid training approach. Similar to Qwen and LLaVA in architecture, it introduces an autoregressive-then-diffusion training strategy:

• Stage 1: Autoregressive fine-tuning for alignment and initial instruction tuning.
• Stage 2: Diffusion-based fine-tuning to enhance instruction-following capabilities.

Trained on the same dataset as LLaVA-NEXT, Dimple-7B outperforms LLaVA-NEXT-7B by 3.9%, indicating that diffusion-based multimodal language models can rival their autoregressive counterparts with a similar training budget.

Model   |    Demo: Chat with Dimple   |    Paper   |    Code  

✨ Features

• Hybrid Training: Integrates autoregressive and diffusion training methods.
• Diffusion Decoding: Supports multiple decoding methods, including confident decoding, random decoding, maskgit-style decoding, and entropy-based decoding.
• Controllable Generation: Allows for fine-grained control over format, structure, and length through structure priors.
• Autoregressive-like Prefilling: Improves inference speed using prefilling techniques.

📊 Evaluation Results

📦 Installation

Make sure your environment includes the following versions:

transformers==4.46.2
torch==2.5.1
accelerate==1.6.0

💻 Usage Examples

Basic Usage

import torch
from transformers import AutoProcessor, AutoModel
import json, requests
from PIL import Image

model_name = "rp-yu/Dimple-7B"
processor = AutoProcessor.from_pretrained(
    model_name,
    trust_remote_code=True
)
model = AutoModel.from_pretrained(
    model_name,
    torch_dtype=torch.bfloat16,
    trust_remote_code=True,
)

image_url = "https://qianwen-res.oss-cn-beijing.aliyuncs.com/Qwen-VL/assets/demo.jpeg"
messages = [
    [{"role": "user", "content": [
        {"type": "image", "image": image_url},
        {"type": "text", "text": "Describe this image."}
    ]}],
]
text = processor.apply_chat_template(
    messages, tokenize=False, add_generation_prompt=True, add_vision_id=False
)
images = [
    Image.open(requests.get(image_url, stream=True).raw).convert("RGB")
]

inputs = processor(
    text=text,
    images=images,
    videos=None,
    padding="longest",
    return_tensors="pt",
)

input_ids = inputs.pop("input_ids")
output = model.diffusion_generate(
    input_ids,
    max_new_tokens=64,
    output_history=True,
    return_dict_in_generate=True,
    steps=64,
    temperature=0.2,
    top_p=0.95,
    alg="origin",
    use_cache=True,
    alg_p_threshold=0.95,
    use_original_confidence=True,
    decoding_pipeline="dim",
    **inputs
)

generations = [
    processor.tokenizer.decode(g[len(p):].cpu().tolist())
    for p, g in zip(input_ids, output.sequences)
]

for j in range(len(messages)):
    print("output:", j, generations[j].split(processor.tokenizer.eos_token)[0])

# output: 0 In the image, a woman wearing a shirt with a plaid and a dog are sitting together on a beach. The sun appears to be setting in the background, creating a warm and serene atmosphere.

📄 License

This project is licensed under the apache-2.0 license.

📚 Citation

@misc{dimple,
      title={Dimple: Discrete Diffusion Multimodal Large Language Model with Parallel Decoding}, 
      author={Runpeng Yu and Xinyin Ma and Xinchao Wang},
      year={2025},
      eprint={2505.16990},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2505.16990}, 
}