diaojiaolou
/
Deep-Live-Cam
mirror of https://github.com/hacksider/Deep-Live-Cam.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Full GPU Utilization

pull/768/head
KRSHH 2024-11-07 10:41:00 +05:30 committed by GitHub
parent 0187936c18
commit 4aadfdf2ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/processors/frame/face_swapper.py
View File

@ -58,17 +58,17 @@ def get_face_swapper() -> Any:
FACE_SWAPPER = insightface.model_zoo.get_model(
model_path, providers=modules.globals.execution_providers
)
# Set optimal inference parameters
# Set optimal inference parameters for maximum VRAM usage
if hasattr(FACE_SWAPPER, "session"):
FACE_SWAPPER.session.set_providers(["CUDAExecutionProvider"])
FACE_SWAPPER.session.set_provider_options(
{
"CUDAExecutionProvider": {
"cuda_mem_limit": 2 * 1024 * 1024 * 1024, # 2GB VRAM limit
"arena_extend_strategy": "kNextPowerOfTwo",
"gpu_mem_limit": 2 * 1024 * 1024 * 1024,
"cudnn_conv_algo_search": "EXHAUSTIVE",
"arena_extend_strategy": "kSameAsRequested", # More aggressive memory allocation
"cudnn_conv_algo_search": "EXHAUSTIVE", # Try all possible algorithms
"do_copy_in_default_stream": True,
"gpu_mem_limit": 0,
"cuda_mem_limit": 0,
}
}
)