sayakpaul HF Staff commited on
Commit
968b96f
·
1 Parent(s): 408c04e
Files changed (1) hide show
  1. 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
- return f
 
 
 
54
  dynamic_shapes=dynamic_shapes,
55
  )
56
  return spaces.aoti_compile(exported, INDUCTOR_CONFIGS)
57
+
58
+ compiled_transformer = f()
59
+ return compiled_transformer