fix load error

vllm/model_executor/models/telechat2.py

@@ -93,6 +93,7 @@ class TeleChat2Model(LlamaModel):
         total_num_heads = self.config.n_head
         head_dim = self.config.hidden_size // total_num_heads
         for name, loaded_weight in weights:
+            if self.use_llama_nn:
                 current_count = loaded_weight.current_count 
                 total_count = loaded_weight.total_count
             if "self_attn.key_value" in name: