fix validation step

export.py

@@ -71,15 +71,11 @@ def convert_openvino(model_id: str, task: str, folder: str) -> List:
             )
             openvino_config = exporter_config_class(model.config)
             inputs = openvino_config.generate_dummy_inputs(framework="pt")
-
             ov_outputs = ov_model(**inputs)
             outputs = model(**inputs)
 
-            if isinstance(outputs, torch.Tensor):
-                outputs = {"logits": outputs}
-                ov_outputs = {"logits": ov_outputs}
-            for output_name in outputs:
-                if not torch.allclose(outputs[output_name], ov_outputs[output_name], atol=1e-3):
+            for output_name in ov_outputs:
+                if isinstance(outputs, torch.Tensor) and not torch.allclose(outputs[output_name], ov_outputs[output_name], atol=1e-3):
                     raise ValueError(
                         "The exported model does not have the same outputs as the original model. Export interrupted."
                     )