[Bugfix] update the prefix for qwen2 (#11795)

Co-authored-by: jiadi.jjd <jiadi.jjd@antgroup.com>
This commit is contained in:
jiangjiadi 2025-01-08 02:36:34 +08:00 committed by GitHub
parent 869579a702
commit c994223d56
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -298,7 +298,7 @@ class Qwen2Model(nn.Module):
lambda prefix: Qwen2DecoderLayer(config=config,
cache_config=cache_config,
quant_config=quant_config,
prefix=f"{prefix}.layers"),
prefix=prefix),
prefix=f"{prefix}.layers",
)