style

Author: IlyasMoutawwakil

optimum/exporters/onnx/model_patcher.py

@@ -357,9 +357,9 @@ def scaled_dot_product_attention(
     scale: Optional[torch._C.Value] = None,
     enable_gqa: bool = False,
 ):
-    assert (not is_causal) or (is_causal and symbolic_helper._is_none(attn_mask)), (
-        "is_causal and attn_mask cannot be set at the same time"
-    )
+    assert (not is_causal) or (
+        is_causal and symbolic_helper._is_none(attn_mask)
+    ), "is_causal and attn_mask cannot be set at the same time"
     assert not enable_gqa, "conversion of scaled_dot_product_attention not implemented if enable_gqa is True"
 
     if symbolic_helper._is_none(scale):