diff --git a/setup.py b/setup.py index 4f9b5961..fc96a5de 100644 --- a/setup.py +++ b/setup.py @@ -15,7 +15,7 @@ from setuptools import setup, find_packages setup( name = 'swarms', packages = find_packages(exclude=[]), - version = '0.2.9', + version = '0.3.0', license='MIT', description = 'Swarms - Pytorch', author = 'Kye Gomez', diff --git a/swarms/agents/workers/models/__init__.py b/swarms/agents/workers/models/__init__.py new file mode 100644 index 00000000..f343a803 --- /dev/null +++ b/swarms/agents/workers/models/__init__.py @@ -0,0 +1,6 @@ +import swarms.agents.workers.models.GroundingDINO.groundingdino.datasets.transforms as T +from swarms.agents.workers.models.GroundingDINO.groundingdino.models import build_model +from swarms.agents.workers.models.GroundingDINO.groundingdino.util import box_ops +from swarms.agents.workers.models.GroundingDINO.groundingdino.util.slconfig import SLConfig +from swarms.agents.workers.models.GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap +from swarms.agents.workers.models.segment_anything.segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator \ No newline at end of file diff --git a/swarms/agents/workers/multi_modal.py b/swarms/agents/workers/multi_modal.py index b3e99658..6271b434 100644 --- a/swarms/agents/workers/multi_modal.py +++ b/swarms/agents/workers/multi_modal.py @@ -33,11 +33,11 @@ from langchain.llms.openai import OpenAI # Grounding DINO # import groundingdino.datasets.transforms as T -import swarms.agents.workers.models.GroundingDINO.groundingdino.datasets.transforms as T -from swarms.agents.workers.models.GroundingDINO.groundingdino.models import build_model -from swarms.agents.workers.models.GroundingDINO.groundingdino.util import box_ops -from swarms.agents.workers.models.GroundingDINO.groundingdino.util.slconfig import SLConfig -from swarms.agents.workers.models.GroundingDINO.groundingdino.util.utils import clean_state_dict, get_phrases_from_posmap +from swarms.agents.workers.models import T +from swarms.agents.workers.models import build_model +from swarms.agents.workers.models import box_ops +from swarms.agents.workers.models import SLConfig +from swarms.agents.workers.models import clean_state_dict, get_phrases_from_posmap from swarms.agents.workers.models.segment_anything.segment_anything import build_sam, SamPredictor, SamAutomaticMaskGenerator import cv2