Spaces:
Running
on
Zero
Running
on
Zero
Update optimization.py
Browse files- 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)
|