Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
cxnt authored Sep 26, 2023
1 parent b73494e commit 219dcba
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions supervisely_integration/docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -7,10 +7,10 @@ RUN pip3 install torchvision==0.13.1+cu113 --extra-index-url https://download.py
RUN pip3 install cachetools
RUN pip3 install timm==0.4.9

RUN python3 -m pip install supervisely==6.72.111
RUN python3 -m pip install supervisely==6.72.137

RUN mkdir -p weights
RUN wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_tiny.pth -O /weights/vit_tiny_sam_hq.pth
RUN wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_b.pth -O /weights/vit_b_sam_hq.pth
RUN wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_l.pth -O /weights/vit_l_sam_hq.pth
RUN wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_h.pth -O /weights/vit_h_sam_hq.pth
RUN wget https://huggingface.co/lkeab/hq-sam/resolve/main/sam_hq_vit_h.pth -O /weights/vit_h_sam_hq.pth

0 comments on commit 219dcba

Please sign in to comment.