How to use dogeplusplus/stable-sam with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("dogeplusplus/stable-sam", dtype=torch.bfloat16, device_map="cuda") prompt = "a photo of samruane cat" image = pipe(prompt).images[0]
from diffusers import StableDiffusionPipeline pipeline = StableDiffusionPipeline.from_pretrained('dogeplusplus/stable-sam') image = pipeline().images[0] image