Lazy cache and debug prints

gradio_demo.py

@@ -131,6 +131,8 @@ with gr.Blocks(
                     inputs=[target_image, texture1, texture2, edit_strength],
                     outputs=blend_output,
                     fn=blend_images,
+                    cache_examples=True,
+                    cache_mode="lazy",
                 )
 
                 blend_btn.click(
@@ -253,6 +255,8 @@ with gr.Blocks(
                     ],
                     outputs=control_output,
                     fn=parametric_control,
+                    cache_examples=True,
+                    cache_mode="lazy",
                 )
 
                 control_btn.click(

marble.py

@@ -43,6 +43,7 @@ def setup_control_mlps(
     ret = {}
     if device is None:
         device = get_device()
+    print(f"Setting up control MLPs on {device}")
     for mlp in CONTROL_MLPS:
         ret[mlp] = setup_control_mlp(mlp, features, device, dtype)
     return ret
@@ -114,7 +115,8 @@ def setup_pipeline(
 ):
     if device is None:
         device = get_device()
-
+        
+    print(f"Setting up pipeline on {device}")
     download_ip_adapter()
 
     cur_block = ("up", 0, 1)