fffiloni commited on
Commit
21e0c98
·
verified ·
1 Parent(s): c603a37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -370,7 +370,7 @@ def warmup_model():
370
  # Generation endpoint — real inference guarded by @spaces.GPU
371
  # ---------------------------------------------------------------------------
372
 
373
- def _gpu_duration(prompt, num_blocks, num_inference_steps, seed):
374
  try:
375
  blocks = int(num_blocks)
376
  steps = int(num_inference_steps)
 
370
  # Generation endpoint — real inference guarded by @spaces.GPU
371
  # ---------------------------------------------------------------------------
372
 
373
+ def _gpu_duration(prompt, num_blocks, num_inference_steps, seed, *args, **kwargs):
374
  try:
375
  blocks = int(num_blocks)
376
  steps = int(num_inference_steps)