Spaces:
Running
on
Zero
Running
on
Zero
Enable aoti
Browse files- optimization.py +1 -1
optimization.py
CHANGED
|
@@ -41,4 +41,4 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
| 41 |
|
| 42 |
pipeline.transformer.fuse_qkv_projections()
|
| 43 |
pipeline.transformer.set_attn_processor(FlashFusedFluxAttnProcessor3_0())
|
| 44 |
-
|
|
|
|
| 41 |
|
| 42 |
pipeline.transformer.fuse_qkv_projections()
|
| 43 |
pipeline.transformer.set_attn_processor(FlashFusedFluxAttnProcessor3_0())
|
| 44 |
+
spaces.aoti_apply(compile_transformer(), pipeline.transformer)
|