[Bugfix] Fix GLM-4.6V vision regression in glm4v_moe and glm_ocr (#20463)
This commit is contained in:
@@ -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]
|
||||
|
||||
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user