The following error occurred.
Please let me know if any process is missing.
Initializing SegmentAnything to cuda:1
╭───────────────────── Traceback (most recent call last) ──────────────────────╮
│ /content/grounded-segment-any-parts/chatbot.py:1337 in │
│ │
│ 1334 │ parser.add_argument('--load', type=str, default="ImageCaptioning_ │
│ 1335 │ args = parser.parse_args() │
│ 1336 │ load_dict = {e.split('')[0].strip(): e.split('')[1].strip() for │
│ ❱ 1337 │ bot = ConversationBot(load_dict=load_dict) │
│ 1338 │ with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as │
│ 1339 │ │ lang = gr.Radio(choices = ['Chinese','English'], value=None, │
│ 1340 │ │ chatbot = gr.Chatbot(elem_id="chatbot", label="CheemsChat") │
│ │
│ /content/grounded-segment-any-parts/chatbot.py:1264 in init │
│ │
│ 1261 │ │ self.models = {} │
│ 1262 │ │ # Load Basic Foundation Models │
│ 1263 │ │ for class_name, device in load_dict.items(): │
│ ❱ 1264 │ │ │ self.models[class_name] = globals()[class_name](device=de │
│ 1265 │ │ │
│ 1266 │ │ # Load Template Foundation Models │
│ 1267 │ │ for class_name, module in globals().items(): │
│ │
│ /content/grounded-segment-any-parts/chatbot.py:1043 in init │
│ │
│ 1040 │ │ │
│ 1041 │ │ sam_checkpoint = "sam_vit_h_4b8939.pth" │
│ 1042 │ │ model_type = "vit_h" │
│ ❱ 1043 │ │ self.sam = sam_model_registry[model_type](checkpoint=sam_chec │
│ 1044 │ │ self.sam.to(device=device) │
│ 1045 │ │
│ 1046 │ @prompts(name="Segment Anything on Image", │
│ │
│ /content/grounded-segment-any-parts/segment-anything/segment_anything/build_ │
│ sam.py:15 in build_sam_vit_h │
│ │
│ 12 │
│ 13 │
│ 14 def build_sam_vit_h(checkpoint=None): │
│ ❱ 15 │ return build_sam( │
│ 16 │ │ encoder_embed_dim=1280, │
│ 17 │ │ encoder_depth=32, │
│ 18 │ │ encoder_num_heads=16, │
│ │
│ /content/grounded-segment-any-parts/segment-anything/segment_anything/build │
│ sam.py:104 in _build_sam │
│ │
│ 101 │ ) │
│ 102 │ sam.eval() │
│ 103 │ if checkpoint is not None: │
│ ❱ 104 │ │ with open(checkpoint, "rb") as f: │
│ 105 │ │ │ state_dict = torch.load(f) │
│ 106 │ │ sam.load_state_dict(state_dict) │
│ 107 │ return sam │
╰──────────────────────────────────────────────────────────────────────────────╯
FileNotFoundError: [Errno 2] No such file or directory: 'sam_vit_h_4b8939.pth'
The following error occurred.
Please let me know if any process is missing.
Initializing SegmentAnything to cuda:1
╭───────────────────── Traceback (most recent call last) ──────────────────────╮
│ /content/grounded-segment-any-parts/chatbot.py:1337 in │
│ │
│ 1334 │ parser.add_argument('--load', type=str, default="ImageCaptioning_ │
│ 1335 │ args = parser.parse_args() │
│ 1336 │ load_dict = {e.split('')[0].strip(): e.split('')[1].strip() for │
│ ❱ 1337 │ bot = ConversationBot(load_dict=load_dict) │
│ 1338 │ with gr.Blocks(css="#chatbot .overflow-y-auto{height:500px}") as │
│ 1339 │ │ lang = gr.Radio(choices = ['Chinese','English'], value=None, │
│ 1340 │ │ chatbot = gr.Chatbot(elem_id="chatbot", label="CheemsChat") │
│ │
│ /content/grounded-segment-any-parts/chatbot.py:1264 in init │
│ │
│ 1261 │ │ self.models = {} │
│ 1262 │ │ # Load Basic Foundation Models │
│ 1263 │ │ for class_name, device in load_dict.items(): │
│ ❱ 1264 │ │ │ self.models[class_name] = globals()[class_name](device=de │
│ 1265 │ │ │
│ 1266 │ │ # Load Template Foundation Models │
│ 1267 │ │ for class_name, module in globals().items(): │
│ │
│ /content/grounded-segment-any-parts/chatbot.py:1043 in init │
│ │
│ 1040 │ │ │
│ 1041 │ │ sam_checkpoint = "sam_vit_h_4b8939.pth" │
│ 1042 │ │ model_type = "vit_h" │
│ ❱ 1043 │ │ self.sam = sam_model_registry[model_type](checkpoint=sam_chec │
│ 1044 │ │ self.sam.to(device=device) │
│ 1045 │ │
│ 1046 │ @prompts(name="Segment Anything on Image", │
│ │
│ /content/grounded-segment-any-parts/segment-anything/segment_anything/build_ │
│ sam.py:15 in build_sam_vit_h │
│ │
│ 12 │
│ 13 │
│ 14 def build_sam_vit_h(checkpoint=None): │
│ ❱ 15 │ return build_sam( │
│ 16 │ │ encoder_embed_dim=1280, │
│ 17 │ │ encoder_depth=32, │
│ 18 │ │ encoder_num_heads=16, │
│ │
│ /content/grounded-segment-any-parts/segment-anything/segment_anything/build │
│ sam.py:104 in _build_sam │
│ │
│ 101 │ ) │
│ 102 │ sam.eval() │
│ 103 │ if checkpoint is not None: │
│ ❱ 104 │ │ with open(checkpoint, "rb") as f: │
│ 105 │ │ │ state_dict = torch.load(f) │
│ 106 │ │ sam.load_state_dict(state_dict) │
│ 107 │ return sam │
╰──────────────────────────────────────────────────────────────────────────────╯
FileNotFoundError: [Errno 2] No such file or directory: 'sam_vit_h_4b8939.pth'