Qwen3.5-397B-A17B-MoE-MXFP4:AMD优化的3970亿参数多模态大模型完整指南
【免费下载链接】Qwen3.5-397B-A17B-MoE-MXFP4项目地址: https://ai.gitcode.com/hf_mirrors/amd/Qwen3.5-397B-A17B-MoE-MXFP4
Qwen3.5-397B-A17B-MoE-MXFP4是一款由AMD优化的3970亿参数多模态大模型,基于Qwen/Qwen3.5-397B-A17B-FP8构建,专为AMD MI350/MI355硬件微架构设计,采用MXFP4量化技术,在保持高性能的同时显著提升了推理效率。
🌟 模型核心特性
🔹 多模态输入支持
- 输入类型:文本、图像、视频
- 输出类型:文本
- 视觉处理:采用27层视觉编码器,支持16×16图像 patch 大小和2×2时空合并(config.json)
🔹 AMD硬件优化
- 支持显卡:AMD MI350 / MI355
- 软件栈要求:
- ROCm 7.2.0
- PyTorch 2.9.1
- Transformers 5.3.0
- 操作系统:Linux
🔹 创新MoE架构
- 专家数量:512个专家
- 每token选择专家数:10个
- 共享专家优化:将共享专家量化为MXFP4并融合到MoE内核(FSE技术),相比BF16减少内存占用并提升解码吞吐量(README.md)
🚀 量化技术解析
🔹 MXFP4量化方案
- 权重量化:OCP MXFP4,静态量化
- 激活量化:OCP MXFP4,动态量化
- 量化工具:AMD-Quark v0.12
- 量化范围:所有MoE专家(包括共享专家)
🔹 量化效果
在GSM8K基准测试中,与原始FP8模型相比:
- 准确率:97.27%(原始模型97.95%)
- 恢复率:99.31%
- 性能提升:显著降低BF16内存占用,提升解码吞吐量(README.md)
💻 快速开始指南
🔹 环境准备
确保系统已安装ROCm 7.2.0和PyTorch 2.9.1,然后克隆仓库:
git clone https://gitcode.com/hf_mirrors/amd/Qwen3.5-397B-A17B-MoE-MXFP4 cd Qwen3.5-397B-A17B-MoE-MXFP4🔹 使用SGLang启动服务
python3 -m sglang.launch_server \ --model-path . \ --tensor-parallel-size 4 \ --trust-remote-code \ --attention-backend aiter \ --mem-fraction-static 0.8 \ --host 0.0.0.0 --port 30000🔹 生成配置参数
默认生成配置(generation_config.json):
- 采样策略:temperature=0.6,top_k=20,top_p=0.95
- Token设置:bos_token_id=248044,eos_token_id=[248046, 248044]
- 最大上下文:支持262144 tokens(config.json)
📊 性能评估
🔹 推理性能
- 部署框架:SGLang
- 并行策略:张量并行(推荐4卡配置)
- 内存优化:mem-fraction-static=0.8(静态内存分配比例)
🔹 评估复现
使用lm-evaluation-harness进行评估:
lm_eval --model local-chat-completions --apply_chat_template \ --tasks gsm8k.yaml \ --num_fewshot 5 \ --model_args "model=.,base_url=http://127.0.0.1:30000/v1/chat/completions,num_concurrent=64" \ --gen_kwargs "max_tokens=12288,temperature=0,top_p=1"📄 许可证信息
- 许可证:Apache-2.0
- 修改声明:© 2026 Advanced Micro Devices, Inc. 保留所有权利
- 原始模型:基于Qwen/Qwen3.5-397B-A17B-FP8构建
🛠️ 技术细节
🔹 模型架构
- 基础架构:Qwen3_5MoeForConditionalGeneration
- 隐藏层大小:4096
- 注意力头数:32(查询头),2(键值头)
- 层数:60层(交替使用线性注意力和全注意力)
🔹 文件结构
- 模型权重:94个分片(model.safetensors-00001-of-00094.safetensors至model.safetensors-00094-of-00094.safetensors)
- 配置文件:config.json(模型架构)、generation_config.json(生成参数)
- 分词器:tokenizer.json、vocab.json、merges.txt
这款由AMD优化的多模态大模型通过创新的MXFP4量化技术和MoE架构设计,为企业级AI应用提供了高性能、高效率的解决方案,特别适合在AMD MI350/MI355平台上部署大规模语言模型应用。
【免费下载链接】Qwen3.5-397B-A17B-MoE-MXFP4项目地址: https://ai.gitcode.com/hf_mirrors/amd/Qwen3.5-397B-A17B-MoE-MXFP4
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考