Using second gpu?

🧨 Diffusers
2

Hi @Telecino! Setting the pipeline’s device to cuda:1 should move it to your second GPU:

from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=True)
pipe = pipe.to("cuda:1")
2 Likes