gabboud commited on
Commit
564b858
·
1 Parent(s): 50f1940

test duration above 2 hours

Browse files
Files changed (1) hide show
  1. utils/pipelines.py +1 -1
utils/pipelines.py CHANGED
@@ -17,7 +17,7 @@ import random
17
  def get_duration(input_file, pdb_file, num_batches, num_designs_per_batch, extra_args, max_duration):
18
  return max_duration
19
 
20
- @spaces.GPU(duration=get_duration)
21
  def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_per_batch, extra_args, max_duration):
22
  """
23
  Runs an unconditional generation with the specified input config file. Saves the generated structures to a timestamped directory in the outputs folder and returns the path to the directory along with a list of the generated structures' file paths.
 
17
  def get_duration(input_file, pdb_file, num_batches, num_designs_per_batch, extra_args, max_duration):
18
  return max_duration
19
 
20
+ @spaces.GPU(duration=7201)
21
  def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_per_batch, extra_args, max_duration):
22
  """
23
  Runs an unconditional generation with the specified input config file. Saves the generated structures to a timestamped directory in the outputs folder and returns the path to the directory along with a list of the generated structures' file paths.