testing Lightning

app.py

@@ -19,7 +19,8 @@ controlnet = ControlNetUnionModel.from_pretrained(
 ).to(device=device)
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16).to(device=device)
 
-model_id = "SG161222/RealVisXL_V5.0"
+#model_id = "SG161222/RealVisXL_V5.0"
+model_id = "SG161222/RealVisXL_V5.0_Lightning"
 pipe = StableDiffusionXLControlNetTileSRPipeline.from_pretrained(
     model_id, controlnet=controlnet, vae=vae, torch_dtype=torch.float16, use_safetensors=True, variant="fp16"
 ).to(device)
@@ -254,6 +255,20 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(), title="MoD ControlNet Tile Upsc
             with gr.Column(scale=12, elem_id="examples_container"):
                 gr.Examples(
                     examples=[
+                        [   "./examples/1.jpg",
+                            prompt.value,
+                            negative_prompt.value,
+                            4096,
+                            0.0,
+                            18,
+                            0.35,
+                            1.0,
+                            0.3,
+                            "LCM",
+                            4,
+                            1024,
+                            "Cosine"
+                        ],
                         [   "./examples/1.jpg",
                             prompt.value,
                             negative_prompt.value,
@@ -267,7 +282,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(), title="MoD ControlNet Tile Upsc
                             4,
                             1024,
                             "Cosine"
-                        ],
+                        ],                   
                         [   "./examples/2.jpg",
                             prompt.value,
                             negative_prompt.value,
@@ -342,7 +357,7 @@ with gr.Blocks(css=css, theme=gr.themes.Ocean(), title="MoD ControlNet Tile Upsc
                     ],                    
                     fn=predict,
                     outputs=result,
-                    cache_examples=True,
+                    cache_examples=False,
                 )
 
     max_tile_size.select(fn=set_maximum_resolution, inputs=[max_tile_size, resolution], outputs=resolution)