[Bugfix] Fix GLM-4.6V vision regression in glm4v_moe and glm_ocr (#20463)

This commit is contained in:
Xinyuan Tong
2026-03-14 04:48:28 +00:00
committed by GitHub
parent dfd0a77a9a
commit c330b687a1
2 changed files with 6 additions and 0 deletions

View File

@@ -281,5 +281,8 @@ class Glm4vMoeForConditionalGeneration(Glm4vForConditionalGeneration):
else:
logger.warning(f"Parameter {name} not found in params_dict")
if not is_nextn:
self.visual.patch_embed.copy_conv3d_weight_to_linear()
EntryClass = [Glm4vMoeForConditionalGeneration]

View File

@@ -431,5 +431,8 @@ class GlmOcrForConditionalGeneration(Glm4vForConditionalGeneration):
)
weight_loader(param, loaded_weight)
if not is_nextn:
self.visual.patch_embed.copy_conv3d_weight_to_linear()
EntryClass = [GlmOcrForConditionalGeneration]