Spaces:
Runtime error
Runtime error
space
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
import sys
|
| 3 |
import time
|
| 4 |
import os
|
|
@@ -41,7 +41,7 @@ UNet3DConditionModel.forward = unet3d_condition_model_forward
|
|
| 41 |
|
| 42 |
MAX_KEYS = 10
|
| 43 |
|
| 44 |
-
|
| 45 |
def core(bundle):
|
| 46 |
generator = torch.Generator().manual_seed(int(bundle["seed"]))
|
| 47 |
result = pipe(
|
|
|
|
| 1 |
+
import spaces
|
| 2 |
import sys
|
| 3 |
import time
|
| 4 |
import os
|
|
|
|
| 41 |
|
| 42 |
MAX_KEYS = 10
|
| 43 |
|
| 44 |
+
@spaces.GPU(duration=120)
|
| 45 |
def core(bundle):
|
| 46 |
generator = torch.Generator().manual_seed(int(bundle["seed"]))
|
| 47 |
result = pipe(
|