Spaces:

xichen98cn
/

FrozenSeg

Runtime error

xichen98cn commited on Sep 7

Commit

c05d999

•

1 Parent(s): ed1e9ad

Update cog.yaml

Files changed (1) hide show

cog.yaml CHANGED Viewed

@@ -22,7 +22,6 @@ build:
     - 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
-    - TORCH_CUDA_ARCH_LIST='7.5' FORCE_CUDA=1 python Mask2Former/mask2former/modeling/pixel_decoder/ops/setup.py build install
 predict: "predict.py:Predictor"

     - 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
+    - TORCH_CUDA_ARCH_LIST='7.5' FORCE_CUDA=1 python frozenseg/modeling/pixel_decoder/ops/setup.py build install
 predict: "predict.py:Predictor"