#!/usr/bin/env python3
import hf_image_uploader as hiu
from diffusers import EulerDiscreteScheduler
from diffusers import AutoPipelineForText2Image
import torch

pipe = AutoPipelineForText2Image.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", variant="fp16", torch_dtype=torch.float16)
pipe.scheduler =  EulerDiscreteScheduler.from_config(pipe.scheduler.config)
pipe.to("cuda")

prompt = "photo of a rhino dressed suit and tie sitting at a table in a bar with a bar stools, award winning photography, Elke vogelsang"
image = pipe(prompt, guidance_scale=9.0, num_inference_steps=25).images[0]

hiu.upload(image, repo_id="patrickvonplaten/images")