cn0303 commited on
Commit
ef62a95
·
verified ·
1 Parent(s): 2bb0dc4

spec parser: GPU duration back to 120s (within ZeroGPU cap)

Browse files
Files changed (1) hide show
  1. spec_brick.py +1 -1
spec_brick.py CHANGED
@@ -100,7 +100,7 @@ if _should_load():
100
  _state["tok"] = tok
101
  _state["model"] = model.to("cuda").eval()
102
 
103
- @spaces.GPU(duration=180) # warm path: load from pre-cached weights + generate
104
  def _generate(text: str) -> str:
105
  if _state["model"] is None:
106
  _load()
 
100
  _state["tok"] = tok
101
  _state["model"] = model.to("cuda").eval()
102
 
103
+ @spaces.GPU(duration=120) # warm path: load from pre-cached weights + generate (120s is within the ZeroGPU cap)
104
  def _generate(text: str) -> str:
105
  if _state["model"] is None:
106
  _load()