stable diffusion test

requirements.txt

torch
diffusers
transformers
pandas

test.py

from diffusers import DiffusionPipeline
import time
import pandas as pd


pipe = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
pipe = pipe.to("cpu")

# Recommended if your computer has < 64 GB of RAM
pipe.enable_attention_slicing()

prompt = "a photo of an astronaut riding a horse on mars"

out = []
for x in range(5):
    start = time.time()
    image = pipe(prompt).images[0]
    out.append(time.time() - start)
image.save('test.png')


df = pd.DataFrame(out)
print(f"mean: {df.mean()[0]}")
print(f"variance: {df.var()[0]}")