[Bugfix][LoRA] Fix missing in_proj_z in Qwen3_5ForConditionalGenerati… (#38927)

Signed-off-by: elenalil-aws <elenalil@amazon.com>
This commit is contained in:
elenalil-aws
2026-04-03 16:30:09 -07:00
committed by GitHub
parent 4b506ff90a
commit 81994e1d0e

View File

@@ -620,6 +620,7 @@ class Qwen3_5ForConditionalGeneration(Qwen3VLForConditionalGeneration, IsHybrid)
self.packed_modules_mapping = {k: list(v) for k, v in base.items()} self.packed_modules_mapping = {k: list(v) for k, v in base.items()}
self.packed_modules_mapping.pop("in_proj_qkvz", None) self.packed_modules_mapping.pop("in_proj_qkvz", None)
self.packed_modules_mapping["in_proj_qkv"] = ["in_proj_qkv"] self.packed_modules_mapping["in_proj_qkv"] = ["in_proj_qkv"]
self.packed_modules_mapping["in_proj_z"] = ["in_proj_z"]
def embed_input_ids( def embed_input_ids(
self, self,