@@ -76,7 +76,7 @@ def _register_composite_model(
7676 model_type = model_type ,
7777 projector_key = projector_key or "multi_modal_projector" ,
7878 vision_model_keys = vision_model_keys or ["vision_tower" ],
79- language_model_keys = language_model_keys or ["language_model" ],
79+ language_model_keys = language_model_keys or ["language_model" , "lm_head" ],
8080 lora_conflict_keys = lora_conflict_keys or [],
8181 )
8282
@@ -200,12 +200,12 @@ def patch_target_modules(
200200
201201
202202_register_composite_model (
203- model_type = "internvl " ,
203+ model_type = "gemma3 " ,
204204)
205205
206206
207207_register_composite_model (
208- model_type = "gemma3 " ,
208+ model_type = "internvl " ,
209209)
210210
211211
@@ -246,14 +246,8 @@ def patch_target_modules(
246246 lora_conflict_keys = ["audio_projection_layer" ],
247247)
248248
249-
250- _register_composite_model (
251- model_type = "paligemma" ,
252- )
253-
254-
255249_register_composite_model (
256- model_type = "video_llava " ,
250+ model_type = "mistral3 " ,
257251)
258252
259253
@@ -264,7 +258,7 @@ def patch_target_modules(
264258
265259
266260_register_composite_model (
267- model_type = "mistral3 " ,
261+ model_type = "paligemma " ,
268262)
269263
270264
@@ -303,3 +297,8 @@ def patch_target_modules(
303297 else ["model" , "lm_head" ],
304298 lora_conflict_keys = ["patch_embed" ],
305299)
300+
301+
302+ _register_composite_model (
303+ model_type = "video_llava" ,
304+ )
0 commit comments