Program not working on GPU but works on CPU

Intermediate
18

I found an old 10x0 related issue…

from diffusers import DiffusionPipeline, AutoencoderKL
import torch

torch.backends.cudnn.benchmark = True # https://github.com/huggingface/diffusers/issues/1556
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16).to("cuda")
pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", vae=vae, variant="fp16", torch_dtype=torch.float16).to("cuda")

prompt = "A clear sunny landscape with mountains and a river"
image = pipe(prompt=prompt).images[0]
image.save("test_image.png")