Update optimization.py

optimization.py

@@ -43,13 +43,13 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
         # dynamic_shapes = tree_map_only((torch.Tensor, bool), lambda x: None, call.kwargs)
         dynamic_shapes |= TRANSFORMER_DYNAMIC_SHAPES
 
-        #quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
+        quantize_(pipeline.transformer, Float8DynamicActivationFloat8WeightConfig())
         
         exported = torch.export.export(
             mod=pipeline.transformer,
             args=call.args,
             kwargs=call.kwargs,
-            dynamic_shapes=dynamic_shapes,
+            # dynamic_shapes=dynamic_shapes,
         )
 
         return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)