Ali2206 commited on
Commit
cbf5c2d
·
verified ·
1 Parent(s): 99c761f

Update src/txagent/txagent.py

Browse files
Files changed (1) hide show
  1. src/txagent/txagent.py +1 -0
src/txagent/txagent.py CHANGED
@@ -11,6 +11,7 @@ import types
11
  from tooluniverse import ToolUniverse
12
  from gradio import ChatMessage
13
  from .toolrag import ToolRAGModel
 
14
 
15
  from .utils import NoRepeatSentenceProcessor, ReasoningTraceChecker, tool_result_format
16
 
 
11
  from tooluniverse import ToolUniverse
12
  from gradio import ChatMessage
13
  from .toolrag import ToolRAGModel
14
+ import torch
15
 
16
  from .utils import NoRepeatSentenceProcessor, ReasoningTraceChecker, tool_result_format
17