fix style

huggingface · Jan 5, 2024 · dbf4a2f · dbf4a2f
1 parent 38a80b9
commit dbf4a2f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/optimum/intel/generation/modeling.py b/optimum/intel/generation/modeling.py
@@ -110,9 +110,7 @@ def __init__(
         self._device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         self.normalized_config = NormalizedConfigManager.get_normalized_config_class(config.model_type)(config)
         self.model_dtype = kwargs.get("model_dtype", None)
-        self.input_names = list(
-            map(lambda x: x.split(".")[0], [inputs.debugName() for inputs in model.graph.inputs()])
-        )
+        self.input_names = [inputs.debugName().split(".")[0] for inputs in model.graph.inputs()]
 
         if is_transformers_version("<=", "4.25.1"):
             self.generation_config = None