cbensimon HF Staff commited on
Commit
7301ed0
·
1 Parent(s): c72054d

Enable aoti

Browse files
Files changed (1) hide show
  1. 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
- # spaces.aoti_apply(compile_transformer(), pipeline.transformer)
 
41
 
42
  pipeline.transformer.fuse_qkv_projections()
43
  pipeline.transformer.set_attn_processor(FlashFusedFluxAttnProcessor3_0())
44
+ spaces.aoti_apply(compile_transformer(), pipeline.transformer)