gabboud commited on
Commit
19021e5
·
1 Parent(s): 564b858

reset dynamic duration, bring back to 1 hour

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. utils/pipelines.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ with gr.Blocks(title="RFD3 Test") as demo:
102
  maximum=10
103
  )
104
  max_duration = gr.Number(
105
- value=240,
106
  label="Max Duration (seconds)",
107
  precision=0,
108
  minimum=1,
 
102
  maximum=10
103
  )
104
  max_duration = gr.Number(
105
+ value=3600,
106
  label="Max Duration (seconds)",
107
  precision=0,
108
  minimum=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=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.
 
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.