linoyts HF Staff commited on
Commit
c654722
·
verified ·
1 Parent(s): 52cbc87

Update optimization.py

Browse files
Files changed (1) hide show
  1. optimization.py +1 -1
optimization.py CHANGED
@@ -37,5 +37,5 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
37
 
38
  return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
39
 
40
- pipeline.transformer.fuse_qkv_projections()
41
  spaces.aoti_apply(compile_transformer(), pipeline.transformer)
 
37
 
38
  return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
39
 
40
+ # pipeline.transformer.fuse_qkv_projections()
41
  spaces.aoti_apply(compile_transformer(), pipeline.transformer)