# expose for downloads

optimization.py

@@ -152,6 +152,10 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
     else:
         compiled_transformer_1, compiled_transformer_2 = compile_transformer()
 
+    # expose for downloads
+    COMPILED_TRANSFORMER_1 = compiled_transformer_1
+    COMPILED_TRANSFORMER_2 = compiled_transformer_2
+
     pipeline.transformer.forward = compiled_transformer_1
     drain_module_parameters(pipeline.transformer)