Skip to content

Commit a433fd7

Browse files
keejkrejclaude
andcommitted
fix: add sam2_weights_filenames for download dialog
Add sam2_weights_filenames to __init__.py so check_model_exists() works correctly for SAM2. This ensures the download warning dialog appears before downloading SAM2 weights, consistent with other models. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent 015a7fc commit a433fd7

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

cellacdc/__init__.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -525,11 +525,18 @@ def printl(*objects, pretty=False, is_decorator=False, idx=1, **kwargs):
525525
]
526526

527527
segment_anything_weights_filenames = [
528-
'sam_vit_h_4b8939.pth',
529-
'sam_vit_l_0b3195.pth',
528+
'sam_vit_h_4b8939.pth',
529+
'sam_vit_l_0b3195.pth',
530530
'sam_vit_b_01ec64.pth'
531531
]
532532

533+
sam2_weights_filenames = [
534+
'sam2.1_hiera_large.pt',
535+
'sam2.1_hiera_base_plus.pt',
536+
'sam2.1_hiera_small.pt',
537+
'sam2.1_hiera_tiny.pt'
538+
]
539+
533540
deepsea_weights_filenames = [
534541
'segmentation.pth',
535542
'tracker.pth'

0 commit comments

Comments
 (0)