gabboud commited on
Commit
e479fab
·
1 Parent(s): c1a83a3

set spaces.GPU for new pipeline

Browse files
Files changed (1) hide show
  1. utils/pipelines.py +1 -0
utils/pipelines.py CHANGED
@@ -98,6 +98,7 @@ def unconditional_generation(num_batches, num_designs_per_batch, length):
98
  except Exception as e:
99
  raise RuntimeError(f"Error during generation: {str(e)}")
100
 
 
101
  def unconditional_generation_with_input_config(input_file):
102
  """
103
  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.
 
98
  except Exception as e:
99
  raise RuntimeError(f"Error during generation: {str(e)}")
100
 
101
+ @spaces.GPU(duration=300)
102
  def unconditional_generation_with_input_config(input_file):
103
  """
104
  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.