def run()

in optimum/commands/export/executorch.py [0:0]

• 18 lines of code
• 5 McCabe index (conditional complexity)

    def run(self):
        from ...exporters.executorch import main_export

        kwargs = {}
        if self.args.use_custom_sdpa:
            kwargs["use_custom_sdpa"] = self.args.use_custom_sdpa
        if self.args.use_custom_kv_cache:
            kwargs["use_custom_kv_cache"] = self.args.use_custom_kv_cache
        if self.args.qlinear:
            kwargs["qlinear"] = self.args.qlinear
        if self.args.qembedding:
            kwargs["qembedding"] = self.args.qembedding

        main_export(
            model_name_or_path=self.args.model,
            task=self.args.task,
            recipe=self.args.recipe,
            output_dir=self.args.output_dir,
            **kwargs,
        )