|
|
|
|
|
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") |
|
|
|
