Update optimization.py

optimization.py

@@ -61,5 +61,5 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
         return aoti_compile(exported, INDUCTOR_CONFIGS)
 
     transformer_config = pipeline.transformer.config
-    pipeline.transformer = cudagraph(compile_transformer())
+    pipeline.transformer = compile_transformer()
     pipeline.transformer.config = transformer_config # pyright: ignore[reportAttributeAccessIssue]