YanZhanPKU/dLLM-PRM-Gap-causal-llada8b

dLLM PRM Gap · Causal LLaDA PRM

💻 Code   •   🤗 Collection   •   📄 Paper

News

• 2026-09: Accepted at NeurIPS 2026.

Public adapter release for dLLM PRM Gap. causal PRM for the cross-backbone LLaDA diagnostic

This repository contains only trainable adapter parameters and the reward head. It does not include the base model. The release configuration is recorded in config.json; the arXiv paper defines the release scope and citation.

Files

• adapter.safetensors — compact adapter weights
• config.json — public base-model id, architecture, and provenance

Load

from huggingface_hub import snapshot_download
from prm.checkpointing import load_diffusion_prm

path = snapshot_download("YanZhanPKU/dLLM-PRM-Gap-causal-llada8b")
model, report = load_diffusion_prm(
    checkpoint=path,
    model_path="GSAI-ML/LLaDA-8B-Base",
    local_files_only=False,
)

Role: cross-backbone diagnostic; it is not a full LLaDA ORM replication.

Paper: https://arxiv.org/abs/2609.35472.