Skip to content

Commit 1a08750

Browse files
authored
fix glm_4_5 (#5347)
1 parent f6f8202 commit 1a08750

File tree

1 file changed

+1
-1
lines changed
  • swift/llm/template/template

1 file changed

+1
-1
lines changed

swift/llm/template/template/glm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,7 @@ def _data_collator_mm_data(self, batch: List[Dict[str, Any]]) -> Dict[str, Any]:
237237

238238
register_template(GLM4_0414TemplateMeta(LLMTemplateType.glm4_0414, template_cls=GLM4_0414Template))
239239

240-
register_template(GLM4TemplateMeta(LLMTemplateType.glm4_5, template_cls=ThinkingTemplate))
240+
register_template(GLM4_0414TemplateMeta(LLMTemplateType.glm4_5, template_cls=ThinkingTemplate))
241241

242242
register_template(GLM4_1VTemplateMeta(MLLMTemplateType.glm4_1v, template_cls=GLM4_1VTemplate))
243243

0 commit comments

Comments
 (0)