KeyArgo
/
smolagents
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix output types with torch not installed

This commit is contained in:
Aymeric 2025-01-10 21:45:19 +01:00
parent 4e80e1b79a
commit 9c1a0fa2e5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/smolagents/types.py
View File

@ -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 # If the class does not have defined output, then we map according to the type
for _k, _v in INSTANCE_TYPE_MAPPING.items(): for _k, _v in INSTANCE_TYPE_MAPPING.items():
if isinstance(output, _k): 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 return output