[model] support keye-vl-8b (#8776)

This commit is contained in:
Kingsley
2025-07-29 21:24:08 +08:00
committed by GitHub
parent ba34fb2860
commit f96e9e434d
4 changed files with 41 additions and 2 deletions

View File

@@ -226,6 +226,15 @@ _register_composite_model(
)
_register_composite_model(
model_type="Keye",
projector_key="mlp_AR",
vision_model_keys=["visual.vision_model.patch_embedding", "visual.vision_model.encoder"],
language_model_keys=["model", "lm_head"],
lora_conflict_keys=["patch_embedding"],
)
_register_composite_model(
model_type="llama4",
vision_model_keys=["vision_model"],