seawolf2357 commited on
Commit
60f80e8
·
verified ·
1 Parent(s): 37e270b

Update optimization.py

Browse files
Files changed (1) hide show
  1. 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)