remove unnecessary names processing

Author: IlyasMoutawwakil

optimum/intel/openvino/modeling.py

@@ -997,20 +997,20 @@ def forward(self, **kwargs):
         np_inputs = isinstance(next(iter(kwargs.values())), np.ndarray)
 
         inputs = {}
-
         for key, value in kwargs.items():
             inputs[key] = np.array(value) if not np_inputs else value
 
-        # Run inference
         outputs = self.request(inputs)
 
         model_outputs = {}
         for key, value in outputs.items():
-            if len(key.names) == 0:
-                key_names = {"no_name_output_O_o"}
+            key_name = next(iter(key.names))
+            if "." in key_name:
+                key_name = key_name.split(".")[0]
+                if key_name not in model_outputs:
+                    model_outputs[key_name] = []
+                model_outputs[key_name].append(torch.from_numpy(value).to(self.device) if not np_inputs else value)
             else:
-                key_names = key.names
-
-            model_outputs[next(iter(key_names))] = torch.from_numpy(value).to(self.device) if not np_inputs else value
+                model_outputs[key_name] = torch.from_numpy(value).to(self.device) if not np_inputs else value
 
         return ModelOutput(**model_outputs)

optimum/intel/openvino/modeling_base.py

@@ -85,10 +85,6 @@ def __init__(
         output_names = {}
         for idx, key in enumerate(model.outputs):
             names = tuple(key.get_names())
-
-            if len(names) == 0:
-                names = ("no_name_output_O_o",)
-
             output_names[next((name for name in names if "/" not in name), names[0])] = idx
         self.output_names = output_names