Fabrice-TIERCELIN commited on
Commit
16ab8c9
·
verified ·
1 Parent(s): f8e6997

@spaces.GPU(duration=1500)

Browse files
Files changed (1) hide show
  1. optimization.py +1 -1
optimization.py CHANGED
@@ -63,7 +63,7 @@ INDUCTOR_CONFIGS = {
63
 
64
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
65
 
66
- #@spaces.GPU(duration=1500)
67
  def compile_transformer():
68
  print("Start compile_transformer()")
69
 
 
63
 
64
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
65
 
66
+ @spaces.GPU(duration=1500)
67
  def compile_transformer():
68
  print("Start compile_transformer()")
69