Spaces:
Running
on
Zero
Running
on
Zero
up
Browse files- optimization.py +3 -1
optimization.py
CHANGED
|
@@ -54,4 +54,6 @@ def compile_transformer(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.k
|
|
| 54 |
dynamic_shapes=dynamic_shapes,
|
| 55 |
)
|
| 56 |
return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
|
| 57 |
-
|
|
|
|
|
|
|
|
|
| 54 |
dynamic_shapes=dynamic_shapes,
|
| 55 |
)
|
| 56 |
return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
|
| 57 |
+
|
| 58 |
+
compiled_transformer = f()
|
| 59 |
+
return compiled_transformer
|