z-lab

/

Alpamayo-R1-10B-DFlash

like 3

Follow

Z Lab 986

This adds the two files needed for diffusers to load the pipeline directly via DFlashPipeline.from_pretrained:

• model_index.json — declares DFlashPipeline as the pipeline class and points
  _target_model_name_or_path to the target/verifier model
• scheduler/scheduler_config.json — config for DFlashTokenDiffusionScheduler

After merging, the pipeline can be loaded with:

import torch
from diffusers import DFlashPipeline

pipe = DFlashPipeline.from_pretrained(
    "z-lab/Alpamayo-R1-10B-DFlash",
    torch_dtype=torch.bfloat16,
    device_map="auto",
)
out = pipe(prompt="Hello, world!")
print(out.texts[0])