Fix CUDA VRAM exhaustion during video processing (#1721)

2026-07-11 04:43:43 +02:00 · 2026-04-02 22:58:47 -04:00
parent 1edc4bc298
commit 11fb5bfbc6
2 changed files with 9 additions and 2 deletions
@@ -108,6 +108,14 @@ def get_face_swapper() -> Any:
                                "MaximumCacheSize": 1024 * 1024 * 512,  # 512MB cache
                            }
                        ))
+                    elif p == "CUDAExecutionProvider":
+                        providers_config.append((
+                            "CUDAExecutionProvider",
+                            {
+                                "arena_extend_strategy": "kSameAsRequested",
+                                "cudnn_conv_algo_search": "DEFAULT",
+                            }
+                        ))
                    else:
                        providers_config.append(p)
                FACE_SWAPPER = insightface.model_zoo.get_model(