Mask2Former

Runtime error

trangvv commited on Aug 15, 2024

Commit

19379d8

verified ·

1 Parent(s): b1030a8

Update cog.yaml

Files changed (1) hide show

cog.yaml CHANGED Viewed

@@ -19,7 +19,7 @@ build:
     - "Cython==0.29.27"
     - "timm==0.4.12"
   run:
-    - pip install detectron2 -f https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.8/index.html
     - pip install git+https://github.com/cocodataset/panopticapi.git
     - pip install git+https://github.com/mcordts/cityscapesScripts.git
     - git clone https://github.com/facebookresearch/Mask2Former

     - "Cython==0.29.27"
     - "timm==0.4.12"
   run:
+    - python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
     - pip install git+https://github.com/cocodataset/panopticapi.git
     - pip install git+https://github.com/mcordts/cityscapesScripts.git
     - git clone https://github.com/facebookresearch/Mask2Former