File size: 658 Bytes
6c27fdd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#!/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")
|