File size: 392 Bytes
139ec19 8a3f1c4 139ec19 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from diffusers import StableDiffusionXLPipeline
import torch
from PIL import Image
import os
pipe = StableDiffusionXLPipeline.from_pretrained(
"stabilityai/sdxl-turbo"
).to("cpu")
pipe.safety_checker = None # Optional
def generate_scene_image(prompt, idx):
image = pipe(prompt).images[0]
out_path = f"assets/images/scene_{idx}.png"
image.save(out_path)
return out_path |