pineedaa/stable-diffusion repository overview

import torch
from diffusers import StableDiffusionPipeline

model_id = "runwayml/stable-diffusion-v1-5"
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
def infer(Prompt):
    image = pipe(Prompt).images[0]  
    
    return image

prompt = "a photo of a cat climbing a tree trying to catch a butterfly"
image = infer(prompt)
image.save("name.png")