Spaces:
Running on Zero
Running on Zero
test duration above 2 hours
Browse files- 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=
|
| 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.
|