Upload 109 files

app.py

@@ -40,37 +40,13 @@ def apply_sage_attention_patch():
         print(msg)
         return msg
 
-def is_fatal_gpu_error(e: BaseException) -> bool:
-    err_type = type(e).__name__
-    err_str = f"{err_type}: {str(e)} | {repr(e)}".lower()
-    fatal_keywords = [
-        "ecc",
-        "uncorrectable",
-        "cudaerror",
-        "acceleratorerror",
-        "cuda error",
-        "cuda_error",
-        "device-side assertion",
-        "cuda_launch_blocking",
-    ]
-    return any(kw in err_str for kw in fatal_keywords)
-
 @spaces.GPU
 def dummy_gpu_for_startup():
-    try:
-        print("--- [GPU Startup] Dummy function for startup check initiated. ---")
-        patch_result = apply_sage_attention_patch()
-        print(f"--- [GPU Startup] {patch_result} ---")
-        print("--- [GPU Startup] Startup check passed. ---")
-        return "Startup check passed."
-    except BaseException as e:
-        if is_fatal_gpu_error(e):
-            print("\n" + "="*80)
-            print(f"🚨 [Fatal GPU Error] Captured fatal GPU error during startup check: {e}")
-            print("🚨 Terminating process to trigger an automatic container restart...")
-            print("="*80 + "\n")
-            os._exit(1)
-        raise e
+    print("--- [GPU Startup] Dummy function for startup check initiated. ---")
+    patch_result = apply_sage_attention_patch()
+    print(f"--- [GPU Startup] {patch_result} ---")
+    print("--- [GPU Startup] Startup check passed. ---")
+    return "Startup check passed."
 
 
 def main():
@@ -97,12 +73,6 @@ def main():
     except BaseException as e:
         err_msg = f"{type(e).__name__}: {str(e)}"
         print(f"--- [GPU Startup] ⚠️ Warning: Startup check failed: {err_msg} ---")
-        if is_fatal_gpu_error(e):
-            print("\n" + "="*80)
-            print(f"🚨 [Fatal GPU Error] Captured fatal GPU error during startup check: {err_msg}")
-            print("🚨 Terminating process to trigger an automatic container restart...")
-            print("="*80 + "\n")
-            os._exit(1)
 
     print("--- Starting Application Setup ---")
     
@@ -120,16 +90,7 @@ def main():
     demo = build_ui(attach_event_handlers)
         
     print("--- Launching Gradio Interface ---")
-    try:
-        demo.queue().launch(server_name="0.0.0.0", server_port=7860)
-    except ValueError as e:
-        if "localhost is not accessible" in str(e):
-            print("\n" + "="*80)
-            print(f"🚨 [Gradio Launch Error] Localhost accessibility check failed: {e}")
-            print("🚨 Terminating process to trigger an automatic container restart...")
-            print("="*80 + "\n")
-            sys.exit(1)
-        raise e
+    demo.queue().launch(server_name="0.0.0.0", server_port=7860)
 
 
 if __name__ == "__main__":

core/pipelines/base_pipeline.py

@@ -38,28 +38,7 @@ class BasePipeline(ABC):
         print(f"Requesting ZeroGPU for {task_name} with duration: {final_duration} seconds.")
         gpu_runner = spaces.GPU(duration=final_duration)(gpu_function)
         
-        try:
-            return gpu_runner(*args, **kwargs)
-        except BaseException as e:
-            err_type = type(e).__name__
-            err_str = f"{err_type}: {str(e)} | {repr(e)}".lower()
-            fatal_keywords = [
-                "ecc",
-                "uncorrectable",
-                "cudaerror",
-                "acceleratorerror",
-                "cuda error",
-                "cuda_error",
-                "device-side assertion",
-                "cuda_launch_blocking",
-            ]
-            if any(kw in err_str for kw in fatal_keywords):
-                print("\n" + "="*80)
-                print(f"🚨 [Fatal GPU Error] Captured fatal GPU error during inference: {e}")
-                print("🚨 Terminating process to trigger an automatic container restart...")
-                print("="*80 + "\n")
-                os._exit(1)
-            raise e
+        return gpu_runner(*args, **kwargs)
 
     def _encode_video_from_frames(self, frames_tensor_cpu: 'torch.Tensor', fps: int, progress: gr.Progress) -> str:
         progress(0.9, desc="Encoding video on CPU...")