#38074 [Model] Add AutoWeightsLoader support for jais
作者 grYe99 · 合并时间 2026-03-25 20:38
为 JAIS 模型添加 load_weights 方法并重构以支持 AutoWeightsLoader,提升一致性。
建议阅读者精读此 PR,以了解如何集成 AutoWeightsLoader 并处理自定义权重逻辑。关注设计决策,如为何保留部分自定义代码而非完全使用 AutoWeightsLoader,这对模型加载模块的设计有参考价值。
参与讨论