mirrors/Fooocus
1
0
Fork 0
Code Issues Pull Requests 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fooocus_version.py
View File

@ -1 +1 @@
version = '2.0.18'
version = '2.0.19'

6
modules/virtual_memory.py
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!')