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,