YiYiXu HF Staff commited on
Commit
8388123
·
verified ·
1 Parent(s): 72c5889

Give each block 6 s of ZeroGPU duration budget (measured ~5 s/block)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 + 4 * int(max_blocks))
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