orronai commited on
Commit
0660e53
·
1 Parent(s): a96a4e2

fix: durations

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.5 if model_type_val == 'FLUX' else 0.15
87
- return (n_max_val * (flowopt_iterations_val + 2) * step_duration) + 30
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(