Spaces:
Running on Zero
Running on Zero
decorate unconditional_generation function
Browse files
app.py
CHANGED
|
@@ -86,6 +86,7 @@ with gr.Blocks(title="RFD3 Test") as demo:
|
|
| 86 |
|
| 87 |
|
| 88 |
# Initialize engine and run generation
|
|
|
|
| 89 |
def unconditional_generation(num_batches, num_designs_per_batch, length):
|
| 90 |
|
| 91 |
config = RFD3InferenceConfig(
|
|
|
|
| 86 |
|
| 87 |
|
| 88 |
# Initialize engine and run generation
|
| 89 |
+
@spaces.GPU(duration=300)
|
| 90 |
def unconditional_generation(num_batches, num_designs_per_batch, length):
|
| 91 |
|
| 92 |
config = RFD3InferenceConfig(
|