feat: format & env

app.py

@@ -123,9 +123,9 @@ def run(
 
     generator = torch.Generator(device=accelerator.device).manual_seed(seed_input)
 
-    if scheduler == 'euler':
+    if scheduler == 'euler' and not isinstance(pipeline.scheduler, FlowMatchEulerDiscreteScheduler):
         pipeline.scheduler = FlowMatchEulerDiscreteScheduler()
-    elif scheduler == 'dpmsolver++':
+    elif scheduler == 'dpmsolver++' and not isinstance(pipeline.scheduler, DPMSolverMultistepScheduler):
         pipeline.scheduler = DPMSolverMultistepScheduler(
             algorithm_type="dpmsolver++",
             solver_type="midpoint",
@@ -201,9 +201,11 @@ title = f"""
 badges_text = r"""
 <div style="text-align: center; display: flex; justify-content: center; gap: 5px;">
 <a href="https://github.com/bytedance/UMO"><img alt="Build" src="https://img.shields.io/github/stars/bytedance/UMO"></a> 
-<a href="https://bytedance.github.io/UMO/"><img alt="Build" src="https://img.shields.io/badge/Project%20Page-UMO-yellow"></a> 
+<a href="https://bytedance.github.io/UMO/"><img alt="Build" src="https://img.shields.io/badge/Project%20Page-UMO-blue"></a> 
+<a href="https://huggingface.co/bytedance-research/UMO"><img src="https://img.shields.io/static/v1?label=%F0%9F%A4%97%20Hugging%20Face&message=Model&color=green"></a>
 <a href="https://arxiv.org/abs/2509.06818"><img alt="Build" src="https://img.shields.io/badge/arXiv%20paper-UMO-b31b1b.svg"></a>
-<a href="https://huggingface.co/bytedance-research/UMO"><img src="https://img.shields.io/static/v1?label=%F0%9F%A4%97%20Hugging%20Face&message=Model&color=orange"></a>
+<a href="https://huggingface.co/spaces/bytedance-research/UMO_UNO"><img src="https://img.shields.io/static/v1?label=%F0%9F%A4%97%20Demo&message=UMO-UNO&color=orange"></a>
+<a href="https://huggingface.co/spaces/bytedance-research/UMO_OmniGen2"><img src="https://img.shields.io/static/v1?label=%F0%9F%A4%97%20Demo&message=UMO-OmniGen2&color=orange"></a>
 </div>
 """.strip()
 
@@ -291,8 +293,8 @@ def main(args):
                     with gr.Row(equal_height=True):
                         global enable_taylorseer
                         global enable_teacache
-                        enable_taylorseer = gr.Checkbox(label="Whether to use TaylorSeer to speed up inference", value=True)
-                        enable_teacache = gr.Checkbox(label="Whether to use TeaCache to speed up inference", value=False)
+                        enable_taylorseer = gr.Checkbox(label="Using TaylorSeer to speed up", value=True)
+                        enable_teacache = gr.Checkbox(label="Using TeaCache to speed up", value=False)
                     
                     with gr.Row(equal_height=True):
                         scheduler_input = gr.Dropdown(

requirements.txt

@@ -1,5 +1,4 @@
-torch==2.6.0
-torchvision==0.21.0
+
 timm
 einops
 accelerate
@@ -18,4 +17,8 @@ ipykernel
 wheel
 triton-windows; sys_platform == "win32"
 # flash-attn==2.7.4.post1
-peft
+peft
+
+--extra-index-url https://download.pytorch.org/whl/cu124
+torch==2.6.0
+torchvision==0.21.0