import torch
from diffusers import StableDiffusionXLPipeline
import os
import secrets

pipe = StableDiffusionXLPipeline.from_pretrained(
    "msArray/animagine-xl-4.0-8bit",
    torch_dtype=torch.float16,
    use_safetensors=True,
    device_map="cuda"
)

prompt = "1girl, toki \(blue archive\), blue archive, safe, masterpiece, high score, great score, absurdres"
negative_prompt = "lowres, bad anatomy, bad hands, text, error, missing finger, extra digits, fewer digits, cropped, worst quality, low quality, low score, bad score, average score, signature, watermark, username, blurry"

image = pipe(
    prompt,
    negative_prompt=negative_prompt,
    width=832,
    height=1216,
    guidance_scale=5,
    num_inference_steps=28
).images[0]

if not os.path.exists("./output"):
    os.makedirs("./output")
    
image.save(f"./output/{secrets.token_hex(10)}.png")