Upload app.py with huggingface_hub

app.py

@@ -50,18 +50,24 @@ processor: Any = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True,
 
 
 def _load_model() -> Any:
-    """SDPA is the reference deployment's attention backend (transformers default).
-
-    The authors' `deploy_hf_liberoplus.py` never requests flash_attention_2, so SDPA is
-    the reference-faithful path — and it avoids CUDA-extension ABI churn entirely.
+    """Load with eager attention.
+
+    The custom `MiBoTForActionGeneration` class declares `_supports_flash_attn` but not
+    `_supports_sdpa`, so under transformers 4.57.1 the only universally available
+    backends are flash_attention_2 and eager. The prebuilt Blackwell flash-attn wheel
+    (pt212) has a torch-ABI mismatch against the runtime's preinstalled torch 2.13
+    (`undefined symbol: ...materialize_cow_storage`), so eager it is — transformers'
+    own suggested fallback. The workload is a single forward pass (~700 vision+text
+    tokens, /no_cot, no autoregressive decoding) + 5 small DiT flow steps, where eager
+    attention costs only a few hundred ms over FA2.
     """
     m: Any = AutoModel.from_pretrained(
         MODEL_ID,
         trust_remote_code=True,
         dtype=torch.bfloat16,
-        attn_implementation="sdpa",
+        attn_implementation="eager",
     )
-    print("[load] attention backend: sdpa (reference default)", flush=True)
+    print("[load] attention backend: eager", flush=True)
     return m.to("cuda")
 
 
@@ -325,9 +331,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="ERVLA — robot action
                     label="Full predicted action chunk",
                     datatype=["number"] * 8,
                 )
-                status_out = gr.Textbox(label="Status", interactive=False, show_copy_button=True)
+                status_out = gr.Textbox(label="Status", interactive=False, buttons=["copy"])
                 csv_out = gr.Textbox(label="Chunk as CSV", lines=8, interactive=False,
-                                     show_copy_button=True)
+                                     buttons=["copy"])
         with gr.Accordion("Advanced", open=False):
             robot_type = gr.Dropdown(
                 choices=LIBERO_ROBOT_TYPES, value="libero_10", label="LIBERO suite (robot type)"