Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

minor cache (#387)

Browse Source
This commit is contained in:
lllyasviel
2023-09-15 15:46:44 -07:00
committed by GitHub
parent 8ef00d87b4
commit 58c29aed00
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/virtual_memory.py
+5 -1
View File
@@ -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!')