Skip to content

feat: Initial block diffusion strategy support - #8

Closed
drewjin with Copilot wants to merge 1 commit into
fast_dllm_v2from
copilot/sub-pr-7
Closed

drewjin with Copilot wants to merge 1 commit into
fast_dllm_v2from
copilot/sub-pr-7

Conversation

Copilot AI commented Dec 10, 2025

Copy link
Copy Markdown

Motivation

This PR introduces a flexible framework to support multiple inference paradigms for Diffusion Language Models (dLLMs). Specifically, it implements a Strategy Registration Mechanism, which decouples the decoding logic from the model architecture. This design allows researchers and developers to rapidly prototype, register, and experiment with novel inference algorithms without modifying the core system.

Implementation Status

Features included in this PR:

  • Strategy Registration Mechanism: A registry system to manage different decoding strategies dynamically.
  • Abstract Strategy Interface: Defined base classes for implementing custom diffusion policies.
  • Initial Block Diffusion Strategy: Implementation of the standard block-wise diffusion logic using the new registry.
  • Model Integration: Basic integration for Fast-dLLM-v2, SDAR, LLaDA-2.0-mini/flash with the new strategy framework.
  • Configuration Support: Ability to select strategies via configuration/arguments.

Features to be implemented in future PRs:

  • Advanced Batching support (leveraging SGLang's core scheduling with new strategies)
  • CUDA Graph optimization for dynamic strategies
  • Support for Prefix Cache with diffusion steps
  • Support for overlap scheduling
  • More advanced decoding policies (e.g., specific low-confidence policies)

Our From-Scratch Diffusion Language Models


💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.

Copilot AI changed the title [WIP] Add initial block diffusion strategy support (feat) Initial block diffusion strategy support Dec 10, 2025
Copilot AI requested a review from drewjin December 10, 2025 13:39
@drewjin drewjin closed this Dec 10, 2025
@drewjin
drewjin deleted the copilot/sub-pr-7 branch December 10, 2025 13:41
@drewjin drewjin changed the title (feat) Initial block diffusion strategy support feat: Initial block diffusion strategy support Dec 25, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants