fix python style checks

intel-analytics · Jun 26, 2024 · 887ac64 · 887ac64
1 parent 9798e05
commit 887ac64
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/llm/src/ipex_llm/transformers/models/qwen2.py b/python/llm/src/ipex_llm/transformers/models/qwen2.py
@@ -418,8 +418,10 @@ def qwen2_attention_forward(
         key_states = self.k_proj(hidden_states)
         value_states = self.v_proj(hidden_states)
         query_states = query_states.view(bsz, q_len, self.num_heads, self.head_dim).transpose(1, 2)
-        key_states = key_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)
-        value_states = value_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim).transpose(1, 2)                                                
+        key_states = key_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim) \
+                               .transpose(1, 2)
+        value_states = value_states.view(bsz, q_len, self.num_key_value_heads, self.head_dim) \
+                                   .transpose(1, 2)
 
     kv_seq_len = key_states.shape[-2]
     if past_key_value is not None: