瀏覽代碼

fix: skip loading extra bias for Qwen2-VL GPTQ (#1040)

AlpinDale 2 月之前
父節點
當前提交
e3f5bae2cc
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      aphrodite/modeling/models/qwen2_vl.py

+ 6 - 0
aphrodite/modeling/models/qwen2_vl.py

@@ -1095,6 +1095,9 @@ class Qwen2VLForConditionalGeneration(nn.Module, SupportsMultiModal):
                 if weight_name not in name:
                     continue
                 name = name.replace(weight_name, param_name)
+                # Skip loading extra bias for GPTQ models.
+                if name.endswith(".bias") and name not in params_dict:
+                    continue
                 param = params_dict[name]
                 weight_loader = param.weight_loader
                 weight_loader(param, loaded_weight, shard_id)
@@ -1119,6 +1122,9 @@ class Qwen2VLForConditionalGeneration(nn.Module, SupportsMultiModal):
                     loaded_weight = loaded_weight.transpose(0, 1)
                     loaded_weight = loaded_weight.reshape(-1)
                 try:
+                    # Skip loading extra bias for GPTQ models.
+                    if name.endswith(".bias") and name not in params_dict:
+                        continue
                     param = params_dict[name]
                 except KeyError:
                     print(params_dict.keys())