requirements.tx

Former-commit-id: 6c93628499
group-chat
Kye 1 year ago
parent a111b99c0c
commit ea9fbfe9c6

@ -1,4 +1,7 @@
faiss-gpu
transformers
pandas
numpy
langchain==0.0.240
nest_asyncio
pegasusx
@ -23,10 +26,6 @@ exxa
open-interpreter
tabulate
colored
# git+https://github.com/IDEA-Research/GroundingDINO.git
# git+https://github.com/facebookresearch/segment-anything.git
addict
albumentations

@ -101,15 +101,10 @@ class OmniModalAgent:
def __init__(
self,
llm: BaseLanguageModel,
tools: List[BaseTool]
# tools: List[BaseTool]
):
self.llm = llm
self.tools = tools
self.chat_planner = load_chat_planner(llm)
self.response_generator = load_response_generator(llm)
self.task_executor: TaskExecutor
def tools_init(self):
self.tools = [
load_tool(tool_name)
for tool_name in [
@ -129,6 +124,10 @@ class OmniModalAgent:
"huggingface-tools/image-transformation",
]
]
self.chat_planner = load_chat_planner(llm)
self.response_generator = load_response_generator(llm)
self.task_executor: TaskExecutor
def run(self, input: str) -> str:

Loading…
Cancel
Save