search

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

Fix: colab demo breaks on cpu runtimes #499

Open aaronrmm opened 1 month ago

aaronrmm commented 1 month ago

In cell 9 the predict method is called without passing the device, and the default is "gpu". This causes the notebook to fail if in a CPU runtime.

The fix is simply to pass in the device that was declared earlier in notebook.

Note: I only tested the changes with a CPU runtime because I can't run predict at all on the free tiers of colab GPU runtimes.