Spaces:
Running on Zero
Running on Zero
set spaces.GPU for new pipeline
Browse files- 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.
|