├── .gitignore ├── .python-version ├── .vscode └── launch.json ├── D2F-eval ├── eval_dream.py ├── eval_dream.sh ├── eval_dream_d2f_vllm.py ├── eval_dream_d2f_vllm.sh ├── eval_llada.py ├── eval_llada.sh ├── generate_llada_demo_ar.py ├── generate_llada_demo_block.py ├── model_cache │ ├── dream │ │ ├── configuration_dream.py │ │ ├── generation_utils.py │ │ └── model_dream.py │ └── llada │ │ ├── configuration_llada.py │ │ └── modeling_llada.py ├── postprocess_code.py └── sanitize.py ├── D2F-train ├── config │ ├── acc_config │ ├── dream_eagle.yaml │ └── llada.yaml ├── model │ ├── configuration_llada.py │ └── modeling_llada.py ├── train.py ├── train.sh └── utils │ ├── __init__.py │ ├── data.py │ ├── generation.py │ ├── loss.py │ ├── model.py │ └── util.py ├── Discrete Diffusion Forcing.pdf ├── LICENCE ├── README.md ├── docs ├── _config.yml ├── assets │ ├── img │ │ └── d2f │ │ │ ├── fig1_main_result.png │ │ │ ├── fig2_tradeoff.png │ │ │ ├── fig3_overview.png │ │ │ ├── fig4_pipeline.png │ │ │ ├── logo_lr.jpg │ │ │ ├── logo_lr.png │ │ │ ├── logo_ud.jpg │ │ │ ├── logo_ud.png │ │ │ ├── table1_llada_results.png │ │ │ ├── table2_dream_results.png │ │ │ └── wechat.png │ └── video │ │ ├── block_demo.mp4 │ │ ├── block_demo_small.mp4 │ │ ├── d2f_vs_ar_demo.mp4 │ │ └── d2f_vs_ar_demo_small.mp4 └── index.html ├── pyproject.toml ├── requirements.txt └── uv.lock /.gitignore: -------------------------------------------------------------------------------- 1 | d2f_vllm 2 | evals_* 3 | *.pyc -------------------------------------------------------------------------------- /.python-version: -------------------------------------------------------------------------------- 1 | 3.12 2 | -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /D2F-eval/eval_dream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_dream.py -------------------------------------------------------------------------------- /D2F-eval/eval_dream.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_dream.sh -------------------------------------------------------------------------------- /D2F-eval/eval_dream_d2f_vllm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_dream_d2f_vllm.py -------------------------------------------------------------------------------- /D2F-eval/eval_dream_d2f_vllm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_dream_d2f_vllm.sh -------------------------------------------------------------------------------- /D2F-eval/eval_llada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_llada.py -------------------------------------------------------------------------------- /D2F-eval/eval_llada.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/eval_llada.sh -------------------------------------------------------------------------------- /D2F-eval/generate_llada_demo_ar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/generate_llada_demo_ar.py -------------------------------------------------------------------------------- /D2F-eval/generate_llada_demo_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/generate_llada_demo_block.py -------------------------------------------------------------------------------- /D2F-eval/model_cache/dream/configuration_dream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/model_cache/dream/configuration_dream.py -------------------------------------------------------------------------------- /D2F-eval/model_cache/dream/generation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/model_cache/dream/generation_utils.py -------------------------------------------------------------------------------- /D2F-eval/model_cache/dream/model_dream.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/model_cache/dream/model_dream.py -------------------------------------------------------------------------------- /D2F-eval/model_cache/llada/configuration_llada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/model_cache/llada/configuration_llada.py -------------------------------------------------------------------------------- /D2F-eval/model_cache/llada/modeling_llada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/model_cache/llada/modeling_llada.py -------------------------------------------------------------------------------- /D2F-eval/postprocess_code.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/postprocess_code.py -------------------------------------------------------------------------------- /D2F-eval/sanitize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-eval/sanitize.py -------------------------------------------------------------------------------- /D2F-train/config/acc_config: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/config/acc_config -------------------------------------------------------------------------------- /D2F-train/config/dream_eagle.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/config/dream_eagle.yaml -------------------------------------------------------------------------------- /D2F-train/config/llada.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/config/llada.yaml -------------------------------------------------------------------------------- /D2F-train/model/configuration_llada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/model/configuration_llada.py -------------------------------------------------------------------------------- /D2F-train/model/modeling_llada.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/model/modeling_llada.py -------------------------------------------------------------------------------- /D2F-train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/train.py -------------------------------------------------------------------------------- /D2F-train/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/train.sh -------------------------------------------------------------------------------- /D2F-train/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /D2F-train/utils/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/utils/data.py -------------------------------------------------------------------------------- /D2F-train/utils/generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/utils/generation.py -------------------------------------------------------------------------------- /D2F-train/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/utils/loss.py -------------------------------------------------------------------------------- /D2F-train/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/utils/model.py -------------------------------------------------------------------------------- /D2F-train/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/D2F-train/utils/util.py -------------------------------------------------------------------------------- /Discrete Diffusion Forcing.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/Discrete Diffusion Forcing.pdf -------------------------------------------------------------------------------- /LICENCE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/LICENCE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/README.md -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/assets/img/d2f/fig1_main_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/fig1_main_result.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/fig2_tradeoff.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/fig2_tradeoff.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/fig3_overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/fig3_overview.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/fig4_pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/fig4_pipeline.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/logo_lr.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/logo_lr.jpg -------------------------------------------------------------------------------- /docs/assets/img/d2f/logo_lr.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/logo_lr.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/logo_ud.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/logo_ud.jpg -------------------------------------------------------------------------------- /docs/assets/img/d2f/logo_ud.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/logo_ud.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/table1_llada_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/table1_llada_results.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/table2_dream_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/table2_dream_results.png -------------------------------------------------------------------------------- /docs/assets/img/d2f/wechat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/img/d2f/wechat.png -------------------------------------------------------------------------------- /docs/assets/video/block_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/video/block_demo.mp4 -------------------------------------------------------------------------------- /docs/assets/video/block_demo_small.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/video/block_demo_small.mp4 -------------------------------------------------------------------------------- /docs/assets/video/d2f_vs_ar_demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/video/d2f_vs_ar_demo.mp4 -------------------------------------------------------------------------------- /docs/assets/video/d2f_vs_ar_demo_small.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/assets/video/d2f_vs_ar_demo_small.mp4 -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/docs/index.html -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/requirements.txt -------------------------------------------------------------------------------- /uv.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhijie-group/Discrete-Diffusion-Forcing/HEAD/uv.lock --------------------------------------------------------------------------------