happyz
/
Fooocus
mirror of https://github.com/lllyasviel/Fooocus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat: add attribute --rebuild-hash-cache, add handling

This commit is contained in:
Manuel Schmid 2024-07-01 17:44:52 +02:00
parent 1fa7eeac84
commit b19ecf410b
No known key found for this signature in database
GPG Key ID: 32C4F7569B40B84B
2 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
args_manager.py
View File

@ -32,7 +32,10 @@ args_parser.parser.add_argument("--enable-describe-uov-image", action='store_tru
help="Disables automatic description of uov images when prompt is empty", default=False)
args_parser.parser.add_argument("--always-download-new-model", action='store_true',
help="Always download newer models ", default=False)
help="Always download newer models", default=False)
args_parser.parser.add_argument("--rebuild-hash-cache", action='store_true',
help="Generates missing model and LoRA hashes.", default=False)
args_parser.parser.set_defaults(
disable_cuda_malloc=True,

14
modules/config.py
View File

@ -757,5 +757,17 @@ def downloading_safety_checker_model():
update_files()
load_cache_from_file()
# write cache to file again for cleanup of invalid cache entries
if args_manager.args.rebuild_hash_cache:
from modules.hash_cache import sha256_from_cache
from modules.util import get_file_from_folder_list
for filename in model_filenames:
filepath = get_file_from_folder_list(filename, paths_checkpoints)
sha256_from_cache(filepath)
for filename in lora_filenames:
filepath = get_file_from_folder_list(filename, paths_loras)
sha256_from_cache(filepath)
# write cache to file again for sorting and cleanup of invalid cache entries
save_cache_to_file()