Speed and UI updates

app.py

@@ -1,17 +1,21 @@
 from diffusers import StableDiffusionXLPipeline
-from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput
+from diffusers.pipelines.stable_diffusion_xl import StableDiffusionXLPipelineOutput, AutoencoderKL
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 import torch
 from PIL import Image, ImageOps
 import gradio as gr
 
+vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
+    vae=vae,
     torch_dtype=torch.float16,
-    variants="fp16",
-    use_safetensor=True,
+    variant="fp16",
+    use_safetensors=True,
 )
 pipe.to("cuda")
+pipe.unet.to(memory_format=torch.channels_last)
+pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
 @torch.no_grad()
 def call(
@@ -320,13 +324,10 @@ with gr.Blocks(css=css) as app:
         '''
         <center>
             <h1>Upside Down Diffusion</h1>
-            <p>Code by Alex Carlier, please follow them on <a href="https://twitter.com/alexcarliera">Twitter</a></p>
+            <p>Code by Alex Carlier, find their <a href="https://colab.research.google.com/drive/1rjDQOn11cTHAf3Oeq87Hfl_Vh41NbTl4?usp=sharing">Google Colab</a> here and please follow them on <a href="https://twitter.com/alexcarliera">Twitter</a></p>
             <p>A space by <a href="https://twitter.com/angrypenguinPNG">AP</a> with contributions from <a href="https://twitter.com/multimodalart">MultimodalArt</a></p>
         </center>
         <hr>
-        <p>
-            🌟 <strong>Create Perception Illusions with Stable Diffusion!</strong> 🌟
-        </p>
         <p>
             Enter your first prompt to craft an image that will show when upright. Then, add a second prompt to reveal a mesmerizing surprise when you flip the image upside down!  ✨
         </p>
@@ -339,8 +340,8 @@ with gr.Blocks(css=css) as app:
     has_flipped = gr.State(value=False)
     with gr.Row():
         with gr.Column():
-            prompt1 = gr.Textbox(label="Prompt 1")
-            prompt2 = gr.Textbox(label="Prompt 2")
+            prompt1 = gr.Textbox(label="Prompt 1", info="Prompt for the side up", placeholder="A sketch of a...")
+            prompt2 = gr.Textbox(label="Prompt 2", info="Prompt for the side down", placeholder="A sketch of a...")
             run_btn = gr.Button("Run")
                 
         with gr.Column():