Fix formatting.

Xilinx · May 31, 2024 · 4e926f9 · 4e926f9
1 parent f624ad5
commit 4e926f9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/brevitas_examples/llm/llm_quant/prepare_for_quantize.py b/src/brevitas_examples/llm/llm_quant/prepare_for_quantize.py
@@ -24,8 +24,7 @@ def _set_bert_mha_attributes(module):
 
 
 _SET_ATTRIBUTES_MAP = {
-    BertAttention: _set_bert_mha_attributes,
-}
+    BertAttention: _set_bert_mha_attributes,}
 
 
 def set_mha_attributes(model):