Don't convert floats or ints to str

huggingface · Jul 2, 2024 · b12c3c9 · b12c3c9
1 parent 87739cd
commit b12c3c9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/transformers/agents/agent_types.py b/src/transformers/agents/agent_types.py
@@ -232,7 +232,7 @@ def to_string(self):
 
 
 AGENT_TYPE_MAPPING = {"text": AgentText, "image": AgentImage, "audio": AgentAudio}
-INSTANCE_TYPE_MAPPING = {str: AgentText, float: AgentText, int: AgentText, Tensor: AgentAudio, ImageType: AgentImage}
+INSTANCE_TYPE_MAPPING = {str: AgentText, Tensor: AgentAudio, ImageType: AgentImage}
 
 
 def handle_agent_inputs(*args, **kwargs):