MrOvkill/stable-diffusion-xl-moddedtohell

1from diffusers import AutoPipelineForText2Image, AutoencoderKL, DiffusionPipeline2import torch3import gradio as gr4 5# vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)6# pipeline_text2image = AutoPipelineForText2Image.from_pretrained(7#     "stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True, revision="fp16", torch_dtype=torch.float16, vae=vae8# )9pipeline_text2image = DiffusionPipeline.from_pretrained(10    "ARDICAI/stable-diffusion-2-1-finetuned"11)12 13def gradio_txt2img(prompt):14    return pipeline_text2image(prompt=prompt).images[0]15 16def api_txt2base64(prompt):17    return gradio_txt2img(prompt).to_base64()18 19with gr.Blocks() as demo:20    gr.Markdown("## Stable Diffusion XL - Modded to Hell Edition")21    gr.Markdown("## Text to Image\n\nConverts a prompt into an image.")22    with gr.Row():23        with gr.Column():24            txt2img_prompt = gr.Textbox(label="Input", lines=2, max_lines=2)25            txt2img_btn = gr.Button("Generate")26        with gr.Column():27            txt2img_out = gr.Image()28    with gr.Row():29        with gr.Column():30            txt2b64_prompt = gr.Textbox(label="Input", lines=2, max_lines=2)31            txt2b64_btn = gr.Button("Generate")32        with gr.Column():33            txt2b64_out = gr.Image()34    txt2img_btn.click(fn=gradio_txt2img, inputs=txt2img_prompt, outputs=txt2img_out, queue=True, api_name="gradio_txt2img")35    txt2b64_btn.click(fn=api_txt2base64, inputs=txt2b64_prompt, outputs=txt2b64_out, queue=True, api_name="gradio_txt2b64")36 37demo.launch(server_name="0.0.0.0", server_port=7860)