Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -406,6 +406,8 @@ def generate(
|
|
| 406 |
|
| 407 |
init_app()
|
| 408 |
|
|
|
|
|
|
|
| 409 |
#Apply the scheduler for the compilation
|
| 410 |
scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=3.0)
|
| 411 |
pipe.scheduler = scheduler
|
|
|
|
| 406 |
|
| 407 |
init_app()
|
| 408 |
|
| 409 |
+
# ==================== AoTI Warmup ====================
|
| 410 |
+
|
| 411 |
#Apply the scheduler for the compilation
|
| 412 |
scheduler = FlowMatchEulerDiscreteScheduler(num_train_timesteps=1000, shift=3.0)
|
| 413 |
pipe.scheduler = scheduler
|