From e36487ffa5b3f03bf6005f186e4784102ad65c6c Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Fri, 16 Aug 2024 04:49:25 -0700 Subject: [PATCH] tune --- backend/memory_management.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/memory_management.py b/backend/memory_management.py index ac57ca68..e7d40200 100644 --- a/backend/memory_management.py +++ b/backend/memory_management.py @@ -497,7 +497,7 @@ def compute_model_gpu_memory_when_using_cpu_swap(current_free_mem, inference_mem k_1GB = max(0.0, min(1.0, k_1GB)) adaptive_safe_factor = 1.0 - 0.23 * k_1GB - suggestion = max(maximum_memory_available * adaptive_safe_factor, maximum_memory_available - 1024 * 1024 * 1024 * 2) + suggestion = max(maximum_memory_available * adaptive_safe_factor, maximum_memory_available - 1024 * 1024 * 1024 * 1.25) return int(max(0, suggestion))