Spaces:
Running on Zero
Running on Zero
reset dynamic duration, bring back to 1 hour
Browse files- app.py +1 -1
- 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=
|
| 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=
|
| 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.
|