Update optimization.py

optimization.py

@@ -89,10 +89,9 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
         compiled_landscape = aoti_compile(exported_landscape, INDUCTOR_CONFIGS)
         print('compiled_landscape', -(t0 - (t0 := datetime.now())))
 
-        compiled_portrait = aoti_compile(exported_portrait, INDUCTOR_CONFIGS)
+        compiled_portrait = aoti_compile(exported_portrait, INDUCTOR_CONFIGS) # TODO: weights_from=compiled_landscape
+        compiled_portrait.weights.clear()
         print('compiled_portrait', -(t0 - (t0 := datetime.now())))
-
-        compiled_portrait.weights = None
         
         return compiled_landscape, compiled_portrait