sirm2024_stablediff/main.py

from diffusers import DiffusionPipeline
import matplotlib.pyplot as plt
import torch

pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16")
pipe.to("cuda")

prompt = "brain magnetic resonance imaging, kandinsky, high resolution, photorealistic"
image = pipe(prompt=prompt).images[0]
plt.imshow(image)