Fix OPT param names (#1819)

2023-11-28 11:22:44 -08:00
parent a1125ad4df
commit b943890484
1 changed files with 3 additions and 0 deletions
--- a/vllm/model_executor/models/opt.py
+++ b/vllm/model_executor/models/opt.py
@@ -331,6 +331,9 @@ class OPTForCausalLM(nn.Module):
                model_name_or_path, cache_dir, load_format, revision):
            if "lm_head.weight" in name:
                continue
+            if name.startswith("decoder."):
+                name = "model." + name
+
            for (param_name, weight_name, shard_id) in stacked_params_mapping:
                if weight_name not in name:
                    continue