[Feature & Fix] Fixed variable usage and added support for CPU devices

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.12k stars 1.31k forks source link

[Feature & Fix] Fixed variable usage and added support for CPU devices #484

Closed ytchen175 closed 2 months ago

ytchen175 commented 2 months ago

[Feature & Fix] Fixed variable usage and added support for CPU devices

Description

Fixed variable usage and added support for CPU devices.

The device variable had not been declared ever
CPU device needs to use float32 to initialize the StableDiffusionInpaintPipeline

Features / Fixes

grounded_sam.ipynb

Impact

grounded_sam.ipynb

rentainhe commented 2 months ago

Thank u so much for this PR!