support pip install

pyproject.toml

@@ -42,7 +42,7 @@ dependencies = [
     "scikit-image==0.21.0",
     "mup==1.0.0",
     "accelerate==0.23.0",
-    "kornia==0.7.0",
+    "kornia==0.6.8",
     "deepspeed==0.10.3",
     "wandb==0.15.12",
     "infinibatch==0.1.1",
@@ -60,7 +60,7 @@ detectron2 = {git = "https://github.com/MaureenZOU/detectron2-xyz.git"}
 "Bug Tracker" = "https://github.com/UX-Decoder/Segment-Everything-Everywhere-All-At-Once/issues"
 
 [tool.setuptools.packages.find]
-exclude = ["assets*"]
+exclude = ["assets*", "demo*"]
 
 [tool.wheel]
-exclude = ["assets*"]
+exclude = ["assets*", "demo*"]

requirements.txt

@@ -0,0 +1,31 @@
+pillow==9.4.0
+opencv-python==4.8.1.78
+pyyaml==6.0.1
+json_tricks==3.17.3
+yacs==0.1.8
+scikit-learn==1.3.1
+pandas==2.0.3
+timm==0.4.12
+numpy==1.23.1
+einops==0.7.0
+fvcore==0.1.5.post20221221
+transformers==4.34.0
+sentencepiece==0.1.99
+ftfy==6.1.1
+regex==2023.10.3
+nltk==3.8.1
+vision-datasets==0.2.2
+cython==3.0.2
+pycocotools==2.0.7
+diffdist==0.1
+pyarrow==13.0.0
+cityscapesscripts==2.2.2
+shapely==1.8.0
+scikit-image==0.21.0
+mup==1.0.0
+accelerate==0.23.0
+kornia==0.7.0
+deepspeed==0.10.3
+wandb
+infinibatch==0.1.1
+gradio==3.42.0

seem/__init__.py

@@ -0,0 +1 @@
+from .modeling import *

seem/modeling/BaseModel.py

@@ -4,7 +4,7 @@ import logging
 import torch
 import torch.nn as nn
 
-from utils.model import align_and_update_state_dicts
+from seem.utils.model import align_and_update_state_dicts
 
 logger = logging.getLogger(__name__)