fix style

intel · Feb 28, 2024 · 2c7315c · 2c7315c
1 parent 136b98c
commit 2c7315c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/llm/src/bigdl/llm/transformers/models/utils.py b/python/llm/src/bigdl/llm/transformers/models/utils.py
@@ -274,7 +274,7 @@ def use_flash_attention(query, key, attention_mask=None):
         else:
             # TODO: below logic may change for different model
             # attention mask shape : [bsz, 1, q_len, k_len]
-            if attention_mask[0].squeeze()[0,0].item() != 0:
+            if attention_mask[0].squeeze()[0, 0].item() != 0:
                 # first batch contains padding
                 # otherwise we suppose it should be a upper triangular matrix
                 # at the same time, the diagonal is also 0