Spaces:
Running
on
Zero
Running
on
Zero
fix: durations
Browse files
app.py
CHANGED
|
@@ -83,8 +83,8 @@ def get_duration(
|
|
| 83 |
src_prompt_val: str, tar_prompt_val: str,
|
| 84 |
src_guidance_scale_val: float, tar_guidance_scale_val: float,
|
| 85 |
) -> float:
|
| 86 |
-
step_duration = 0.
|
| 87 |
-
return (n_max_val * (flowopt_iterations_val + 2) * step_duration) +
|
| 88 |
|
| 89 |
@spaces.GPU(duration=get_duration)
|
| 90 |
def FlowOpt_run(
|
|
|
|
| 83 |
src_prompt_val: str, tar_prompt_val: str,
|
| 84 |
src_guidance_scale_val: float, tar_guidance_scale_val: float,
|
| 85 |
) -> float:
|
| 86 |
+
step_duration = 0.6 if model_type_val == 'FLUX' else 0.2
|
| 87 |
+
return (n_max_val * (flowopt_iterations_val + 2) * step_duration) + 45
|
| 88 |
|
| 89 |
@spaces.GPU(duration=get_duration)
|
| 90 |
def FlowOpt_run(
|