Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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)
|