try to solve some of the GPU Task Aborted for stable aud

app.py

@@ -109,10 +109,13 @@ def get_duration(input_audio,
 
  duration = min(utils.get_duration(input_audio), utils.MAX_DURATION)
  time_for_maxlength = factor * forwards * 0.15 # 0.25 is the time per forward pass
- print('expected time:', time_for_maxlength / utils.MAX_DURATION * duration)
-
+ 
+ if model_id != STABLEAUD:
+ time_for_maxlength = time_for_maxlength / utils.MAX_DURATION * duration
+ 
+ print('expected time:', time_for_maxlength)
  spare_time = 5
- return max(10, time_for_maxlength / utils.MAX_DURATION * duration + spare_time)
+ return max(10, time_for_maxlength + spare_time)
 
 
 def verify_model_params(model_id: str, input_audio, src_prompt: str, tar_prompt: str, cfg_scale_src: float,