Skip to content

Commit 1e78f0a

Browse files
author
Sayan Shaw
committed
fix subscripting error
1 parent 9e637c1 commit 1e78f0a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnxruntime_extensions/_hf_cvt.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -301,7 +301,8 @@ def post_processing(self, **kwargs):
301301
op_class = SingleOpGraph.get_op_class(_cvt_op)
302302
default_inputs = op_class.input_default_values()
303303
if default_inputs is None:
304-
if self.cvt_quadruple.default_encoder_inputs["fairseq"]:
304+
encoder_inputs = self.cvt_quadruple.default_encoder_inputs
305+
if encoder_inputs is not None and encoder_inputs["fairseq"]:
305306
pass
306307
else:
307308
return g

0 commit comments

Comments
 (0)