Fix ModelPatcher returns empty outputs (#2109)

* fix bug `ModelPatcher` returns empty outputs When model's output is tuple or list, `filtered_outputs` doesn't get assigned and hence always a empty dict * typo --------- Co-authored-by: Ella Charlaix <[email protected]>
huggingface · Dec 2, 2024 · 3ba1057 · 3ba1057
1 parent f22655c
commit 3ba1057
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/exporters/onnx/model_patcher.py b/optimum/exporters/onnx/model_patcher.py
@@ -168,7 +168,7 @@ def patched_forward(*args, **kwargs):
                         filterd_outputs[name] = value
             elif isinstance(outputs, (list, tuple)):
                 outputs_list = list(config.outputs.keys())
-                dict(zip(outputs_list, outputs))
+                filterd_outputs = dict(zip(outputs_list, outputs))
             else:
                 if len(config.outputs) > 1:
                     num_outputs = len(config.outputs)