From 9c1a0fa2e58e99b41503f8a80f060de847c4b28d Mon Sep 17 00:00:00 2001 From: Aymeric Date: Fri, 10 Jan 2025 21:45:19 +0100 Subject: [PATCH] Fix output types with torch not installed --- src/smolagents/types.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/smolagents/types.py b/src/smolagents/types.py index a9730c1..f579e5a 100644 --- a/src/smolagents/types.py +++ b/src/smolagents/types.py @@ -277,7 +277,8 @@ def handle_agent_output_types(output, output_type=None): # If the class does not have defined output, then we map according to the type for _k, _v in INSTANCE_TYPE_MAPPING.items(): if isinstance(output, _k): - return _v(output) + if not isinstance(output, object): # avoid converting to audio if torch is not installed + return _v(output) return output