update dev env

Author: FynnBe

dev/env.yaml

@@ -40,9 +40,10 @@ dependencies:
   - segment-anything # for model testing
   - tensorflow>=2,<3
   - timm # for model testing
-  - torchvision
+  - torchvision>=0.21
   - tqdm
   - typing-extensions
   - xarray
   - pip:
+      - git+https://github.com/ChaoningZhang/MobileSAM.git # for model testing
       - -e ..

setup.py

@@ -46,7 +46,9 @@
     ],
     include_package_data=True,
     extras_require={
-        "pytorch": (pytorch_deps := ["torch>=1.6,<3", "torchvision", "keras>=3.0,<4"]),
+        "pytorch": (
+            pytorch_deps := ["torch>=1.6,<3", "torchvision>=0.21", "keras>=3.0,<4"]
+        ),
         "tensorflow": ["tensorflow", "keras>=2.15,<4"],
         "onnx": ["onnxruntime"],
         "dev": (