IDEA-Research / Grounded-Segment-Anything

Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
https://arxiv.org/abs/2401.14159
Apache License 2.0
14.11k stars 1.31k forks source link

Seeing error when running the gradio app #459

Closed jloveric closed 4 months ago

jloveric commented 4 months ago

I see the following error when running the gradio app through docker

File "/opt/conda/lib/python3.10/site-packages/torch/serialization.py", line 282, in __init__
    super(_open_zipfile_reader, self).__init__(torch._C.PyTorchFileReader(name_or_buffer))
RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory

The docker file appears to run and build fine. In addition I can bring up the gradio app, but when I try to use the UI this is what I see after executing the text prompt.

jloveric commented 4 months ago

Ok, it may be that sam_vit_h_4b8939.pth is not completely downloaded, looks like it's only 400mb instead of ~2gb

jloveric commented 4 months ago

Yep, the .pth files weren't fully downloaded