Skip to content

Commit 83dc433

Browse files
committed
fix 0.3b
1 parent e96befb commit 83dc433

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lightllm/models/vit/layer_weights/transformer_layer_weight.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,10 +149,10 @@ def load_hf_weights(self, weights):
149149
weights[self._v_bias_name] = v_bias_
150150
del weights[f"vision_model.encoder.layers.{self.layer_num_}.attn.qkv.bias"]
151151

152-
if self._q_norm_weight_name in weights:
152+
if self.qk_norm and self._q_norm_weight_name in weights:
153153
weights[self._q_norm_weight_name] = F.pad(weights[self._q_norm_weight_name], (0, self.padding_hidden_size))
154154

155-
if self._k_norm_weight_name in weights:
155+
if self.qk_norm and self._k_norm_weight_name in weights:
156156
weights[self._k_norm_weight_name] = F.pad(weights[self._k_norm_weight_name], (0, self.padding_hidden_size))
157157

158158
if f"vision_model.encoder.layers.{self.layer_num_}.ls1" in weights:

0 commit comments

Comments
 (0)