duration 960 second

utils/pipelines.py

@@ -101,7 +101,7 @@ def unconditional_generation(num_batches, num_designs_per_batch, length):
     except Exception as e:
         raise RuntimeError(f"Error during generation: {str(e)}")
     
-@spaces.GPU(duration=3600) 
+@spaces.GPU(duration=960) 
 def generation_with_input_config(input_file, pdb_file, num_batches, num_designs_per_batch, extra_args):
     """
     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.