Spaces:
Paused
Paused
Update optimization.py
Browse files- optimization.py +1 -1
optimization.py
CHANGED
|
@@ -129,4 +129,4 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
|
|
| 129 |
drain_module_parameters(pipeline.transformer)
|
| 130 |
|
| 131 |
pipeline.transformer_2.forward = combined_transformer_2
|
| 132 |
-
drain_module_parameters(pipeline.transformer_2)
|
|
|
|
| 129 |
drain_module_parameters(pipeline.transformer)
|
| 130 |
|
| 131 |
pipeline.transformer_2.forward = combined_transformer_2
|
| 132 |
+
drain_module_parameters(pipeline.transformer_2)
|