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)
|