From 58c29aed0077c9355ee05aa7aa72c8532cfd6cef Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Fri, 15 Sep 2023 15:46:44 -0700 Subject: [PATCH] minor cache (#387) --- fooocus_version.py | 2 +- modules/virtual_memory.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/fooocus_version.py b/fooocus_version.py index fc45298..d37de15 100644 --- a/fooocus_version.py +++ b/fooocus_version.py @@ -1 +1 @@ -version = '2.0.18' +version = '2.0.19' diff --git a/modules/virtual_memory.py b/modules/virtual_memory.py index 799ff18..3c590c0 100644 --- a/modules/virtual_memory.py +++ b/modules/virtual_memory.py @@ -85,7 +85,11 @@ def move_to_virtual_memory(model, comfy_unload=True): filename = model_file['filename'] prefix = model_file['prefix'] - safetensors_keys = only_load_safetensors_keys(filename) + if hasattr(model, 'virtual_memory_safetensors_keys'): + safetensors_keys = model.virtual_memory_safetensors_keys + else: + safetensors_keys = only_load_safetensors_keys(filename) + model.virtual_memory_safetensors_keys = safetensors_keys if safetensors_keys is None: print(f'[Virtual Memory System] Error: The Virtual Memory System currently only support safetensors models!')