sayakpaul HF Staff commited on
Commit
c341814
·
1 Parent(s): 6a0340f
Files changed (1) hide show
  1. optimization.py +3 -3
optimization.py CHANGED
@@ -100,9 +100,9 @@ def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kw
100
 
101
  compiled_landscape = aoti_compile(exported_landscape, INDUCTOR_CONFIGS)
102
  compiled_portrait = aoti_compile(exported_portrait, INDUCTOR_CONFIGS)
103
- compiled_portrait.weights = (
104
- compiled_landscape.weights
105
- ) # Avoid weights duplication when serializing back to main process
106
 
107
  return compiled_landscape, compiled_portrait
108
 
 
100
 
101
  compiled_landscape = aoti_compile(exported_landscape, INDUCTOR_CONFIGS)
102
  compiled_portrait = aoti_compile(exported_portrait, INDUCTOR_CONFIGS)
103
+ # compiled_portrait.weights = (
104
+ # compiled_landscape.weights
105
+ # ) # Avoid weights duplication when serializing back to main process
106
 
107
  return compiled_landscape, compiled_portrait
108