
NVIDIA Efficient-DLM-4B革命性扩散语言模型如何实现并行文本生成【免费下载链接】Efficient-DLM-4B项目地址: https://ai.gitcode.com/hf_mirrors/nvidia/Efficient-DLM-4BEfficient-DLM-4B是一款专为并行生成设计的基础扩散语言模型它通过高效的连续预训练将预训练的自回归语言模型AR LMs转换为扩散语言模型在保持强大AR模型任务准确性的同时实现了更快的解码速度。模型概述从AR到扩散模型的突破Efficient-DLM-4B的核心创新在于其独特的架构设计它采用块式注意力block-wise attention与干净上下文条件化技术实现了KV缓存友好的解码过程。这种设计使得模型能够在生成文本时并行处理多个块大幅提升了生成效率。此外模型还引入了位置相关的令牌掩蔽position-dependent token masking技术有效减少了扩散生成中的训练-测试不匹配问题。这些技术的结合使得Efficient-DLM-4B在保持高生成质量的同时实现了显著的速度提升。核心技术解析块式扩散注意力机制Efficient-DLM-4B的Qwen3FlexAttention类实现了一种创新的块式扩散注意力机制。这种机制结合了三种关键掩码块对角掩码M_BD用于噪声块内的自注意力偏移块因果掩码M_OBC用于条件上下文的交叉注意力块因果掩码M_BC用于更新x0的注意力这种复合掩码设计使得模型能够在训练和推理过程中灵活地控制信息流平衡生成质量和速度。高效的令牌掩蔽策略模型实现了两种令牌掩蔽策略基础掩蔽和扩展掩蔽。扩展掩蔽策略采用两阶段损坏过程支持按块采样通过指数加权函数实现了后期重的采样模式这有助于模型更好地捕捉序列中的依赖关系。并行生成流程Efficient-DLM-4B的generate方法实现了基于块扩散的并行生成流程。通过将生成过程分为多个步骤在每个步骤中并行处理多个块模型能够显著加速文本生成过程。用户可以通过调整steps和block_length参数来平衡生成速度和质量。环境配置与快速开始要使用Efficient-DLM-4B首先需要确保安装了正确版本的transformers库transformers4.52.2然后可以通过以下步骤克隆仓库并开始使用git clone https://gitcode.com/hf_mirrors/nvidia/Efficient-DLM-4B cd Efficient-DLM-4B与Efficient-DLM-4B对话以下是使用Efficient-DLM-4B进行对话的简单示例from transformers import AutoModel, AutoTokenizer import torch repo_name nvidia/Efficient-DLM-4B tokenizer AutoTokenizer.from_pretrained(repo_name, trust_remote_codeTrue) model AutoModel.from_pretrained(repo_name, trust_remote_codeTrue) model model.cuda().to(torch.bfloat16) user_input input(User: ).strip() prompt_ids tokenizer(user_input, return_tensorspt).input_ids.to(devicecuda) out_ids, nfe model.generate( prompt_ids, max_new_tokens128, steps128, block_length32, shift_logitsFalse, temperature0.7, threshold0.9, ) response tokenizer.batch_decode(out_ids[:, prompt_ids.shape[1]:], skip_special_tokensTrue)[0] print(fModel: {response}) print(f[Num Function Eval (NFE){nfe}])性能优势Efficient-DLM-4B在准确性和吞吐量之间取得了优异的平衡。通过并行生成技术它能够在保持与传统AR模型相当的生成质量的同时大幅提升生成速度。这种性能优势使得Efficient-DLM-4B特别适合需要快速响应的应用场景如实时对话系统、智能客服等。引用如果您在研究中使用了Efficient-DLM-4B请引用以下论文article{fu2025efficient, title{Efficient-dlm: From autoregressive to diffusion language models, and beyond in speed}, author{Fu, Yonggan and Whalen, Lexington and Ye, Zhifan and Dong, Xin and Diao, Shizhe and Liu, Jingyu and Wu, Chengyue and Zhang, Hao and Xie, Enze and Han, Song and others}, journal{arXiv preprint arXiv:2512.14067}, year{2025} }Efficient-DLM-4B代表了语言模型生成效率的重要突破通过创新性的扩散机制和并行处理技术为下一代高性能NLP应用开辟了新的可能性。无论是学术研究还是工业应用这款模型都为开发者提供了一个强大而高效的工具助力构建更快、更智能的语言处理系统。【免费下载链接】Efficient-DLM-4B项目地址: https://ai.gitcode.com/hf_mirrors/nvidia/Efficient-DLM-4B创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考