nsfwalex commited on
Commit
221414e
·
1 Parent(s): a78ded8

perf(zerogpu): cap generate_image GPU slot at duration=45 for better scheduling

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -975,7 +975,7 @@ def _anima_step_progress(pipe, callback, total_steps):
975
  sched.step = orig
976
 
977
 
978
- @spaces.GPU
979
  def generate_image(
980
  model_name,
981
  prompt,
 
975
  sched.step = orig
976
 
977
 
978
+ @spaces.GPU(duration=45)
979
  def generate_image(
980
  model_name,
981
  prompt,