|
|
|
@ -20,6 +20,8 @@ from langchain.tools import BaseTool
|
|
|
|
|
from pydantic import Field
|
|
|
|
|
|
|
|
|
|
from swarms.utils.logger import logger
|
|
|
|
|
from langchain.tools.file_management.write import WriteFileTool
|
|
|
|
|
from langchain.tools.file_management.read import ReadFileTool
|
|
|
|
|
|
|
|
|
|
llm = ChatOpenAI(model_name="gpt-4", temperature=1.0)
|
|
|
|
|
|
|
|
|
@ -164,7 +166,6 @@ def compile(task: str):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# mm model workers
|
|
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
import torch
|
|
|
|
|
from PIL import Image
|
|
|
|
|