Skip to content

Commit a6f5839

Browse files
committed
we need to pass the manger not the env
1 parent 3cb321d commit a6f5839

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

.github/scripts/install_model.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from java.lang import System
33
from java.util.function import Consumer
44
from ai.nets.samj.ij.utils import Constants
5+
from ai.nets.samj.install import Sam2EnvManager
56

67
from io.bioimage.modelrunner.system import PlatformDetection
78

@@ -18,8 +19,8 @@ def accept(self, s):
1819
else:
1920
relative_mamba = "appose_" + PlatformDetection.getArch()
2021

21-
22-
model = SAM2Tiny(os.path.join(Constants.FIJI_FOLDER, relative_mamba))
22+
manager = Sam2EnvManager(os.path.join(Constants.FIJI_FOLDER, relative_mamba), "tiny")
23+
model = SAM2Tiny(manager)
2324
model.getInstallationManger().setConsumer(PrintConsumer());
2425
model.getInstallationManger().installEverything();
2526

0 commit comments

Comments
 (0)