add deepseekv3 and llama4

This commit is contained in:
Chranos
2026-02-11 14:30:01 +08:00
parent 8ac7afcbd3
commit 96ed925486
2 changed files with 31 additions and 5 deletions

View File

@@ -389,6 +389,15 @@ def vllm__llama4__Llama4ForCausalLM__load_weights(
if "rotary_emb.inv_freq" in name:
continue
# Strip language_model. prefix for Llama4ForConditionalGeneration
if name.startswith("language_model."):
name = name[len("language_model."):]
# Skip vision encoder weights
elif (name.startswith("multi_modal_projector.")
or name.startswith("vision_encoder.")
or name.startswith("vision_model.")):
continue
# Permute Q/K weights for rotary embedding
name, loaded_weight = self.permute_qk_weight_for_rotary(
name, loaded_weight)