Spaces:
Running
on
Zero
Running
on
Zero
up
Browse files- 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 |
-
|
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 |
|