ovi054 commited on
Commit
fb1983e
·
verified ·
1 Parent(s): 7d594f1

Update optimization.py

Browse files
Files changed (1) hide show
  1. optimization.py +1 -1
optimization.py CHANGED
@@ -40,7 +40,7 @@ INDUCTOR_CONFIGS = {
40
 
41
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
42
 
43
- @spaces.GPU()
44
  def compile_transformer():
45
 
46
  # pipeline.load_lora_weights(
 
40
 
41
  def optimize_pipeline_(pipeline: Callable[P, Any], *args: P.args, **kwargs: P.kwargs):
42
 
43
+ @spaces.GPU(duration=1200)
44
  def compile_transformer():
45
 
46
  # pipeline.load_lora_weights(