Skip to content

Commit 1a582c6

Browse files
authored
Pass all CLI args into main_export (#153)
1 parent ae2d32a commit 1a582c6

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

optimum/commands/export/executorch.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,16 @@ def run(self):
147147
kwargs["disable_dynamic_shapes"] = self.args.disable_dynamic_shapes
148148
if self.args.qlinear:
149149
kwargs["qlinear"] = self.args.qlinear
150+
if self.args.qlinear_group_size:
151+
kwargs["qlinear_group_size"] = self.args.qlinear
152+
if self.args.qlinear_encoder:
153+
kwargs["qlinear_encoder"] = self.args.qlinear
154+
if self.args.qlinear_encoder_group_size:
155+
kwargs["qlinear_encoder_group_size"] = self.args.qlinear
150156
if self.args.qembedding:
151157
kwargs["qembedding"] = self.args.qembedding
158+
if self.args.qembedding_group_size:
159+
kwargs["qembedding_group_size"] = self.args.qembedding
152160
if self.args.max_seq_len:
153161
kwargs["max_seq_len"] = self.args.max_seq_len
154162

0 commit comments

Comments
 (0)