Explorar o código

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

AlpinDale hai 2 meses
pai
achega
e3f5bae2cc
Modificáronse 1 ficheiros con 6 adicións e 0 borrados
  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())