From 23a7559db84865a9d448b60fdfb16dc8858d9395 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Sun, 22 Oct 2023 12:16:36 -0700 Subject: [PATCH] Fooocus GitHub Bot Commit This commit is generated by a GitHub bot of Fooocus --- backend/headless/fcbh/model_management.py | 6 +++++- fooocus_version.py | 2 +- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/backend/headless/fcbh/model_management.py b/backend/headless/fcbh/model_management.py index d6cee6c..75108ee 100644 --- a/backend/headless/fcbh/model_management.py +++ b/backend/headless/fcbh/model_management.py @@ -339,7 +339,11 @@ def free_memory(memory_required, device, keep_loaded=[]): if unloaded_model: soft_empty_cache() - + else: + if vram_state != VRAMState.HIGH_VRAM: + mem_free_total, mem_free_torch = get_free_memory(device, torch_free_too=True) + if mem_free_torch > mem_free_total * 0.25: + soft_empty_cache() def load_models_gpu(models, memory_required=0): global vram_state diff --git a/fooocus_version.py b/fooocus_version.py index 08913c7..024a3d1 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.1.731' +version = '2.1.732'