Skip to content

Commit d7cf486

Browse files
committed
fix dep issue
1 parent f633b15 commit d7cf486

4 files changed

Lines changed: 8 additions & 4 deletions

File tree

.env.example

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,5 @@ OPENAI_API_KEY=your-api-key
22
AZURE_OPENAI_URL=
33
OLLAMA_HOST=http://ollama.server:11434
44
# do not change this TORCH_HOME variable
5-
TORCH_HOME=./pretrained_models
5+
TORCH_HOME=./pretrained_models
6+
HF_HOME=./pretrained_models/hf

naver/tool/sam.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,4 +41,5 @@ def forward(self, image: np.ndarray, bbox, use_image_patch_coord: bool = True) -
4141
@classmethod
4242
def prepare(cls):
4343
path = get_root_folder() / "pretrained_models" / "sam"
44-
N.util.download_file("https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth", str(path))
44+
if not (path / "sam_vit_h_4b8939.pth").exists():
45+
N.util.download_file("https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth", str(path))

pixi.toml

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,8 @@ scipy = ">=1.15.3,<2"
3737
accelerate = ">=1.8.0,<2"
3838
sentencepiece = ">=0.2.0,<0.3"
3939
orjson = ">=3.10.14,<4"
40+
fsspec = ">=2023.5.0, <2025.5.0"
41+
mkl = ">=2023.1.0, <2024.0.0"
4042

4143
[pypi-dependencies]
4244
transformers = "~=4.43.0"
@@ -48,7 +50,7 @@ ollama = ">=0.5.1, <0.6"
4850
problog = ">=2.2.7, <3"
4951
tensorneko = "==0.3.21"
5052
timm = ">=1.0.16, <1.1"
51-
hydra_vl4ai = "==0.0.5"
53+
hydra_vl4ai = "==0.0.6"
5254
scallopy = { git = "https://github.com/scallop-lang/scallop.git", subdirectory = "etc/scallopy", rev = "f8fac18" }
5355
segment_anything = { path = "module_repos/Grounded-Segment-Anything/segment_anything" }
5456
flash_attn = { url = "https://github.com/ControlNet/flash-attention-prebuild-wheels/releases/download/v0.0.7/flash_attn-2.7.4.post1+cu118torch2.5-cp312-cp312-linux_x86_64.whl" }

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@
5050
"problog",
5151
"timm",
5252
"tensorneko==0.3.21",
53-
"hydra_vl4ai==0.0.5",
53+
"hydra_vl4ai==0.0.6",
5454
],
5555
classifiers=[
5656
"License :: OSI Approved :: Apache Software License",

0 commit comments

Comments
 (0)