fix bug

app.py

@@ -259,7 +259,7 @@ with gr.Blocks() as demo:
             preset_dropdown = gr.Dropdown(choices=list(
                 presets.keys()) + ["Custom"], label="Choose Preset", value=default_preset)
             nfe_slider = gr.Slider(4, 64, value=len(
-                cache_schedule['attn']), step=1, label="Number of Steps (NFE)")
+                cache_schedule_state['attn']), step=1, label="Number of Steps (NFE)")
         with gr.Group():
             gr.Markdown(
                 "Click Grid to Customize Cache Schedule<br>🟧 = Compute Attn Layer <br> 🟩 = Compute FFN Layer <br> ⬜ = Cached Layer", container=True)