lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 18:51:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

simplify

Browse Source
This commit is contained in:
layerdiffusion
2024-08-16 04:55:01 -07:00
parent e36487ffa5
commit 394da01959
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/memory_management.py
View File

@@ -493,11 +493,10 @@ def free_memory(memory_required, device, keep_loaded=[]):
def compute_model_gpu_memory_when_using_cpu_swap(current_free_mem, inference_memory): def compute_model_gpu_memory_when_using_cpu_swap(current_free_mem, inference_memory):
maximum_memory_available = current_free_mem - inference_memory maximum_memory_available = current_free_mem - inference_memory
k_1GB = float(inference_memory / (1024 * 1024 * 1024)) suggestion = max(
k_1GB = max(0.0, min(1.0, k_1GB)) maximum_memory_available / 1.3,
maximum_memory_available - 1024 * 1024 * 1024 * 1.25
adaptive_safe_factor = 1.0 - 0.23 * k_1GB )
suggestion = max(maximum_memory_available * adaptive_safe_factor, maximum_memory_available - 1024 * 1024 * 1024 * 1.25)
return int(max(0, suggestion)) return int(max(0, suggestion))