Skip to content

Commit b7ab7ff

Browse files
authored
Update docker-compose.yml
- Upgrade to SAM 2.1
1 parent f9cf1e9 commit b7ab7ff

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

label_studio_ml/examples/segment_anything_2_video/docker-compose.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,9 +24,9 @@ services:
2424
# specify device
2525
- DEVICE=cuda # or 'cpu' (coming soon)
2626
# SAM2 model config
27-
- MODEL_CONFIG=sam2_hiera_l.yaml
27+
- MODEL_CONFIG=configs/sam2.1/sam2.1_hiera_l.yaml
2828
# SAM2 checkpoint
29-
- MODEL_CHECKPOINT=sam2_hiera_large.pt
29+
- MODEL_CHECKPOINT=sam2.1_hiera_large.pt
3030

3131
# Specify the Label Studio URL and API key to access
3232
# uploaded, local storage and cloud storage files.

0 commit comments

Comments
 (0)