def ovmodel_kwargs()

in optimum_benchmark/backends/openvino/backend.py [0:0]

• 15 lines of code
• 7 McCabe index (conditional complexity)

    def ovmodel_kwargs(self) -> Dict[str, Any]:
        kwargs = {}

        if self.config.export is not None:
            kwargs["export"] = self.config.export

        if self.config.use_cache is not None:
            kwargs["use_cache"] = self.config.use_cache

        if self.config.use_merged is not None:
            kwargs["use_merged"] = self.config.use_merged

        if self.config.load_in_8bit is not None:
            kwargs["load_in_8bit"] = self.config.load_in_8bit

        if self.config.load_in_4bit is not None:
            kwargs["load_in_4bit"] = self.config.load_in_4bit

        if self.config.ov_config:
            kwargs["ov_config"] = self.config.ov_config

        return kwargs