Spaces:
Running
Running
fix bug
Browse files
app.py
CHANGED
@@ -258,8 +258,7 @@ with gr.Blocks() as demo:
|
|
258 |
with gr.Column(scale=0):
|
259 |
preset_dropdown = gr.Dropdown(choices=list(
|
260 |
presets.keys()) + ["Custom"], label="Choose Preset", value=default_preset)
|
261 |
-
nfe_slider = gr.Slider(4, 64, value=
|
262 |
-
cache_schedule_state['attn']), step=1, label="Number of Steps (NFE)")
|
263 |
with gr.Group():
|
264 |
gr.Markdown(
|
265 |
"Click Grid to Customize Cache Schedule<br>π§ = Compute Attn Layer <br> π© = Compute FFN Layer <br> β¬ = Cached Layer", container=True)
|
|
|
258 |
with gr.Column(scale=0):
|
259 |
preset_dropdown = gr.Dropdown(choices=list(
|
260 |
presets.keys()) + ["Custom"], label="Choose Preset", value=default_preset)
|
261 |
+
nfe_slider = gr.Slider(4, 64, value=32), step=1, label="Number of Steps (NFE)")
|
|
|
262 |
with gr.Group():
|
263 |
gr.Markdown(
|
264 |
"Click Grid to Customize Cache Schedule<br>π§ = Compute Attn Layer <br> π© = Compute FFN Layer <br> β¬ = Cached Layer", container=True)
|