执行摘要

本PR修复了Grok-1 AMD INT4模型加载时因缺失rope_theta属性导致的崩溃问题。通过将grok.py中的get_rope_config调用替换为本地属性提取逻辑，恢复了之前的行为并支持Transformers v5。变更仅影响AMD特定的Grok-1 INT4模型，解决了CI测试失败。

功能与动机

AMD夜间CI测试nightly-8-gpu-grok1-int4和nightly-8-gpu-mi35x-grok1-int4在加载Grok-1 INT4模型时崩溃，错误为AttributeError: 'Grok1Config' object has no attribute 'rope_theta'。这是由于PR #21135将grok.py从安全的getattr(config, "rope_theta", 10000)迁移到共享的get_rope_config(config) helper函数，而Grok-1 INT4模型使用自定义Grok1Config类，未定义rope_theta属性。

实现拆解

只修改了python/sglang/srt/models/grok.py文件：

• 移除对get_rope_config的导入。
• 在类的__init__方法中，将rope_theta, _ = get_rope_config(config)替换为本地逻辑：

rope_theta = getattr(config, "rope_theta", None)
if rope_theta is None:
    rope_params = getattr(config, "rope_parameters", None)
    rope_theta = rope_params["rope_theta"] if rope_params else 10000

这首先尝试v4路径的rope_theta，然后v5路径的rope_parameters["rope_theta"]，最后默认10000。

评论区精华

Review讨论较少，reviewer yctseng0211直接批准。在Issue评论中，作者michaelzhang-ai强调了变更范围有限："no NVIDIA / common code impact"，并确认了Grok-1 FP8和Grok-2不受影响。yctseng0211更新分支以包含PR 21547解决stage-a failure。

风险与影响

• 风险：逻辑简单，回归风险低；但依赖于rope_parameters字典的存在，如果其他模型配置类似缺失，可能未覆盖。未添加单元测试，长期维护可能脆弱。
• 影响：仅修复Grok-1 AMD INT4模型加载，使相关CI测试通过。无其他模型或代码路径受影响，因为grok.py是Grok-1专用。

关联脉络

本PR直接关联到#21135，后者引入了导致问题的变更。从近期历史PR看，涉及AMD的PR如#21657优化了MoEGate性能，但本PR专注于bugfix，显示了在硬件特定模型配置处理上的持续改进。