gpu_duration

backend.py

@@ -100,6 +100,7 @@ def _duration_for(
     mode: str,
     preset: str,
     multiplier: float = 1.0,
+    gpu_duration: int = -1,
     progress: Any = None,
 ) -> int:
     """ZeroGPU duration estimator. Same signature as _execute_workflow.
@@ -115,6 +116,9 @@ def _duration_for(
     needed for style + lipsync detailer paths). If the server rejects values
     in this range, the user will see a clear error and can retry.
     """
+    if gpu_duration != -1:
+        return gpu_duration
+        
     base = _BASE_DURATION_S.get(mode, 180)
     mult = _PRESET_MULT.get(preset.lower(), 1.5)
     frames = _frames_from_workflow(workflow)
@@ -138,6 +142,7 @@ def _execute_workflow(
     mode: str,
     preset: str,
     multiplier: float = 1.0,
+    gpu_duration: int = -1,
     progress: Any = None,
 ) -> str:
     """Run the workflow on GPU and return the path of the first video output.
@@ -385,7 +390,7 @@ class ComfyUILibraryBackend:
         *,
         preset: str = "balanced",
         duration_multiplier: float = 1.0,
-        gpu_duration: int = 0,  # legacy, ignored (now derived from preset+frames)
+        gpu_duration: int = -1,
         progress: Any = None,
     ) -> AsyncIterator[Any]:
         """Run a workflow end-to-end. Yields Download/Progress/Output/Error events.
@@ -467,7 +472,7 @@ class ComfyUILibraryBackend:
                 # light calls get fast queue priority while heavy ones reserve
                 # real headroom. Off-Spaces it's a plain call.
                 video_path = _execute_workflow(
-                    self._executor, workflow, output_ids, mode, preset, duration_multiplier, progress,
+                    self._executor, workflow, output_ids, mode, preset, duration_multiplier, gpu_duration, progress,
                 )
                 # Fallback: if history_result didn't surface a path (rare on
                 # Spaces — happens when ZeroGPU's subprocess boundary drops