mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-03-02 09:46:00 +08:00
[model] support GLM-OCR SFT (#10183)
This commit is contained in:
@@ -239,6 +239,15 @@ _register_composite_model(
|
||||
)
|
||||
|
||||
|
||||
_register_composite_model(
|
||||
model_type="glm_ocr",
|
||||
projector_key="visual.merger",
|
||||
vision_model_keys=["visual.patch_embed", "visual.blocks"],
|
||||
language_model_keys=["language_model", "lm_head"],
|
||||
lora_conflict_keys=["patch_embed"],
|
||||
)
|
||||
|
||||
|
||||
_register_composite_model(
|
||||
model_type="internvl",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user