hotfix transformers version 4.41 (#447)

modelscope · May 20, 2024 · 28e6db6 · 28e6db6
1 parent bc3a365
commit 28e6db6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/modelscope_agent/tools/hf_tool.py b/modelscope_agent/tools/hf_tool.py
@@ -2,7 +2,11 @@
 
 import json
 from modelscope_agent.tools.base import BaseTool, register_tool
-from transformers.tools import Tool as HFTool
+
+try:
+    from transformers.tools import Tool as HFTool
+except ImportError:
+    from transformers import Tool as HFTool
 
 
 @register_tool('hf-tool')