diffusers/tools

1#!/usr/bin/env python32from diffusers import DiffusionPipeline, AutoencoderKL3import hf_image_uploader as hiu4import torch5 6vae = AutoencoderKL.from_pretrained(7    "madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float168)9pipe = DiffusionPipeline.from_pretrained(10    "stabilityai/stable-diffusion-xl-base-1.0",11    variant="fp16",12    vae=vae,13    torch_dtype=torch.float16,14).to("cuda")15 16pipe.load_lora_weights("rvorias/m_test")17# pipe.enable_xformers_memory_efficient_attention()18 19image = pipe("a photo of a pikachu pixel art", generator=torch.manual_seed(66)).images[0]20hiu.upload(image, "patrickvonplaten/images")21