From 57ae757757387eebd765b9f8a1806de7ada07c6c Mon Sep 17 00:00:00 2001 From: eaidova Date: Wed, 4 Dec 2024 09:08:42 +0400 Subject: [PATCH] fix code style --- optimum/exporters/openvino/utils.py | 10 +++++++++- optimum/intel/openvino/modeling_base.py | 12 ++++++------ 2 files changed, 15 insertions(+), 7 deletions(-) diff --git a/optimum/exporters/openvino/utils.py b/optimum/exporters/openvino/utils.py index 5242db1d1a..9891395a38 100644 --- a/optimum/exporters/openvino/utils.py +++ b/optimum/exporters/openvino/utils.py @@ -216,7 +216,15 @@ def get_submodels(model): return custom_export, fn_get_submodels -MULTI_MODAL_TEXT_GENERATION_MODELS = ["llava", "llava-next", "llava-qwen2", "internvl-chat", "minicpmv", "phi3-v", "qwen2-vl"] +MULTI_MODAL_TEXT_GENERATION_MODELS = [ + "llava", + "llava-next", + "llava-qwen2", + "internvl-chat", + "minicpmv", + "phi3-v", + "qwen2-vl", +] def save_config(config, save_dir): diff --git a/optimum/intel/openvino/modeling_base.py b/optimum/intel/openvino/modeling_base.py index d5d5666891..75c3f4b3c3 100644 --- a/optimum/intel/openvino/modeling_base.py +++ b/optimum/intel/openvino/modeling_base.py @@ -111,9 +111,9 @@ def __init__( for idx, key in enumerate(model.inputs): names = tuple(key.get_names()) input_names[next((name for name in names if "/" not in name), names[0])] = idx - input_dtypes[ - next((name for name in names if "/" not in name), names[0]) - ] = key.get_element_type().get_type_name() + input_dtypes[next((name for name in names if "/" not in name), names[0])] = ( + key.get_element_type().get_type_name() + ) self.input_names = input_names self.input_dtypes = input_dtypes @@ -122,9 +122,9 @@ def __init__( for idx, key in enumerate(model.outputs): names = tuple(key.get_names()) output_names[next((name for name in names if "/" not in name), names[0])] = idx - output_dtypes[ - next((name for name in names if "/" not in name), names[0]) - ] = key.get_element_type().get_type_name() + output_dtypes[next((name for name in names if "/" not in name), names[0])] = ( + key.get_element_type().get_type_name() + ) self.output_names = output_names self.output_dtypes = output_dtypes