#!/usr/bin/env python3
from diffusers import DiffusionPipeline, AutoencoderKL
import hf_image_uploader as hiu
import torch

vae = AutoencoderKL.from_pretrained(
    "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
)
pipe = DiffusionPipeline.from_pretrained(
    "stabilityai/stable-diffusion-xl-base-1.0",
    variant="fp16",
    vae=vae,
    torch_dtype=torch.float16,
).to("cuda")

pipe.load_lora_weights("rvorias/m_test")
# pipe.enable_xformers_memory_efficient_attention()

image = pipe("a photo of a pikachu pixel art", generator=torch.manual_seed(66)).images[0]
hiu.upload(image, "patrickvonplaten/images")