Skip to content

Commit c0bb9aa

Browse files
lint fix
1 parent 740a940 commit c0bb9aa

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxscript/rewriter/ort_fusions/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,7 @@ def rewrite(
211211
Dh_q = self.bindings.get("Dh_q")
212212
Dh_k = self.bindings.get("Dh_k")
213213
Dh_v = self.bindings.get("Dh_v")
214-
qkv_hidden_sizes=[Dh_q, Dh_k, Dh_v]
214+
qkv_hidden_sizes = [Dh_q, Dh_k, Dh_v]
215215

216216
if self._has_past:
217217
attention, present = op.Attention(

0 commit comments

Comments
 (0)