reduce GPU time to 240s

utils/pipelines.py

@@ -10,7 +10,7 @@ from utils.handle_files import *
 import sys
 import shutil
 
-@spaces.GPU(duration=300)  
+@spaces.GPU(duration=240)  
 def test_rfd3_from_notebook():
     # Set seed for reproducibility
     seed_everything(0)
@@ -44,7 +44,7 @@ def test_rfd3_from_notebook():
     
 
 # Initialize engine and run generation
-@spaces.GPU(duration=300) 
+@spaces.GPU(duration=240) 
 def unconditional_generation(num_batches, num_designs_per_batch, length):
     """
     Runs an unconditional generation with the specified parameters for number of batches, number of designs per batch, and length of the generated proteins. 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.
@@ -99,7 +99,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=300) 
+@spaces.GPU(duration=240) 
 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.