title

app-img2img.py

@@ -49,7 +49,7 @@ else:
 pipe.set_progress_bar_config(disable=True)
 pipe.to(torch_device="cuda", torch_dtype=torch.float32)
 pipe.unet.to(memory_format=torch.channels_last)
-# pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
+pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 user_queue_map = {}
 
 # for torch.compile
@@ -145,7 +145,7 @@ async def stream(user_id: uuid.UUID):
     try:
         user_queue = user_queue_map[uid]
         queue = user_queue["queue"]
-        
+
         async def generate():
             while True:
                 data = await queue.get()

img2img/index.html

@@ -166,12 +166,12 @@
     </div>
     <main class="container mx-auto px-4 py-4 max-w-4xl flex flex-col gap-4">
         <article class="text-center max-w-xl mx-auto">
-            <h1 class="text-3xl font-bold  mb-4">Real-Time Latent Consistency Model</h1>
-
+            <h1 class="text-3xl font-bold">Real-Time Latent Consistency Model</h1>
+            <h2 class="text-2xl font-bold mb-4">Image to Image</h2>
             <p class="text-sm">
                 This demo showcases
                 <a href="https://huggingface.co/SimianLuo/LCM_Dreamshaper_v7" target="_blank"
-                    class="text-blue-500 hover:underline">LCM</a>
+                    class="text-blue-500 hover:underline">LCM</a> Image to Image pipeline
                 using
                 <a href="https://github.com/huggingface/diffusers/tree/main/examples/community#latent-consistency-pipeline"
                     target="_blank" class="text-blue-500 hover:underline">Diffusers</a> with a MJPEG