Update optimization.py

optimization.py

@@ -123,8 +123,8 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
     transformer_dtype = pipeline.transformer.dtype
     cache_context = pipeline.transformer.cache_context
 
-    with torch.no_grad():
-        combined_transformer(**call.kwargs)
+    # with torch.no_grad():
+    #     combined_transformer(**call.kwargs)
 
     pipeline.transformer = combined_transformer
     # pipeline.transformer = cudagraph(combined_transformer)