Skip to content

Commit c1a9ca9

Browse files
author
Sanggyu Lee
committed
Remove layer_idx from op_attention inputs
1 parent 09fd22a commit c1a9ca9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tico/serialize/operators/onert/op_attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ def define_node(
143143
circle.BuiltinOperator.BuiltinOperator.ATTENTION, self._op_codes
144144
)
145145

146-
inputs = node.args
146+
inputs = node.args[:-1] # all args except for last (= layer_idx)
147147
outputs = [node]
148148
operator = create_builtin_operator(self.graph, op_index, inputs, outputs)
149149

0 commit comments

Comments
 (0)