Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -709,9 +709,7 @@ def get_gpu_duration(
|
|
| 709 |
num_inference_steps: int = 8,
|
| 710 |
progress=None,
|
| 711 |
):
|
| 712 |
-
|
| 713 |
-
step_ratio = num_inference_steps / 8 # Normalize to 8 steps as baseline
|
| 714 |
-
return int(base_duration * step_ratio)
|
| 715 |
|
| 716 |
@spaces.GPU(duration=get_gpu_duration)
|
| 717 |
@torch.inference_mode()
|
|
|
|
| 709 |
num_inference_steps: int = 8,
|
| 710 |
progress=None,
|
| 711 |
):
|
| 712 |
+
return int(gpu_duration)
|
|
|
|
|
|
|
| 713 |
|
| 714 |
@spaces.GPU(duration=get_gpu_duration)
|
| 715 |
@torch.inference_mode()
|