reproducible stablediffusion using seeds

stablediffusion_reproducible_seeds.py

import torch
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=True).to("cuda")


seed= 111111
height =512
width = 512
device = 'cuda'


generator = torch.Generator(device=device)
generator = generator.manual_seed(seed)
latents = torch.randn(
        (1, pipe.unet.in_channels, height // 8, width // 8),
        generator = generator,
        device = device
)

p =  pipe("fruits basket",
          latents=latents)

img=p.images[0]
img