victor HF Staff commited on
Commit
84a2935
·
1 Parent(s): 588ebcf

Reduce GPU duration to 30s (saves quota)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -63,7 +63,7 @@ async def homepage():
63
  return f.read()
64
 
65
 
66
- @spaces.GPU
67
  def run_inference(prompt, width, height, seed):
68
  """Run the diffusion pipeline on GPU."""
69
  generator = torch.Generator("cuda").manual_seed(seed)
 
63
  return f.read()
64
 
65
 
66
+ @spaces.GPU(duration=30)
67
  def run_inference(prompt, width, height, seed):
68
  """Run the diffusion pipeline on GPU."""
69
  generator = torch.Generator("cuda").manual_seed(seed)