Skip to content

Commit f154a5b

Browse files
JingyaHuangdacorvo
authored andcommitted
fix: default shapes
1 parent f461b7e commit f154a5b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

optimum/neuron/utils/argument_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -90,6 +90,7 @@ class InputShapesArguments:
9090
feature_size: int | None = None
9191
nb_max_frames: int | None = None
9292
audio_sequence_length: int | None = None
93+
visual_seq_length: int | None = None
9394
point_batch_size: int | None = None
9495
nb_points_per_image: int | None = None
9596
num_beams: int | None = None

0 commit comments

Comments
 (0)