Spaces:
Running on Zero
Running on Zero
Give each block 6 s of ZeroGPU duration budget (measured ~5 s/block)
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ def stop():
|
|
| 52 |
return "stopping after this block…"
|
| 53 |
|
| 54 |
|
| 55 |
-
@spaces.GPU(duration=lambda image, prompt, max_blocks: 30 +
|
| 56 |
def rollout(image, prompt, max_blocks):
|
| 57 |
held = CONTROLS["▲ forward (W)"]
|
| 58 |
|
|
|
|
| 52 |
return "stopping after this block…"
|
| 53 |
|
| 54 |
|
| 55 |
+
@spaces.GPU(duration=lambda image, prompt, max_blocks: 30 + 6 * int(max_blocks))
|
| 56 |
def rollout(image, prompt, max_blocks):
|
| 57 |
held = CONTROLS["▲ forward (W)"]
|
| 58 |
|