jkim96's picture
Add DASH-Q remote-code inference (Triton decode kernel)
e4037f2 verified
|
Raw
History Blame
2.08 kB
metadata
license: other
license_name: exaone
base_model: LGAI-EXAONE/EXAONE-4.5-33B
base_model_relation: quantized
library_name: transformers
tags:
  - dashq
  - quantized
  - post-training-quantization
  - int4

DASH-Q

EXAONE-4.5-33B-DASHQ-INT4-g64

DASH-Q — Diagonal-Aware Shrinkage for Robust PTQ. INT4 · group size 64 · 23.1319 GB (from 68.7003 GB — 3.0x smaller)

Usage

from transformers import AutoModelForImageTextToText, AutoTokenizer

model = AutoModelForImageTextToText.from_pretrained(
    "jkim96/EXAONE-4.5-33B-DASHQ-INT4-g64", trust_remote_code=True, device_map="cuda", dtype="auto"
)
tokenizer = AutoTokenizer.from_pretrained("jkim96/EXAONE-4.5-33B-DASHQ-INT4-g64")

messages = [{"role": "user", "content": "Explain 2-bit quantization in one sentence."}]
text = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
inputs = tokenizer(text, return_tensors="pt").to(model.device)
print(tokenizer.decode(model.generate(**inputs, max_new_tokens=256)[0]))

trust_remote_code=True is required: the checkpoint ships its quantized-layer implementation (modeling_dashq.py) and Triton kernels (dashq_kernel.py). Without Triton, or on CPU, it falls back to dequantize-and-matmul in PyTorch.

Requirements

Package Minimum Verified with
torch 2.4 2.12.1+cu130
transformers 5.8 5.9.0
triton 3.0 (Linux; bundled with CUDA builds of PyTorch) 3.7.1
huggingface_hub 1.5 (pulled in by transformers) 1.15.0

Quantization

Field Value
Base model LGAI-EXAONE/EXAONE-4.5-33B
Precision INT4, group size 64
Scale / zero dtype float16
Calibration wikitext2, 128 samples x 2048
Size 23.1319 GB · original 68.7003 GB · 3.0x compression

Benchmarks

Full zero-shot / few-shot results for every DASH-Q checkpoint: github.com/JaeminK/dashq#benchmarks