Unverified Commit 63baa28c authored by Zyyeric's avatar Zyyeric Committed by GitHub
Browse files

[Model] Enable LoRA support for tower and connector in GLM4-V (#31652)


Signed-off-by: default avatarZyyeric <eric1976808123@gmail.com>
Co-authored-by: default avatarJee Jee Li <pandaleefree@gmail.com>
parent e5173d3b
......@@ -1788,6 +1788,20 @@ class Glm4vForConditionalGeneration(
tower_model="visual.",
)
def get_num_mm_encoder_tokens(
self,
num_image_tokens: int,
) -> int:
merge_size = self.config.vision_config.spatial_merge_size
return num_image_tokens * (merge_size**2)
def get_num_mm_connector_tokens(
self,
num_vision_tokens: int,
) -> int:
merge_size = self.config.vision_config.spatial_merge_size
return num_vision_tokens // (merge_size**2)
@MULTIMODAL_REGISTRY.register_processor(
Glm4vMultiModalProcessor,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment