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

feat: only use hash cache logs for exceptions

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

2
modules/config.py
View File

@ -762,12 +762,14 @@ if args_manager.args.rebuild_hash_cache:
from modules.hash_cache import sha256_from_cache
from modules.util import get_file_from_folder_list
print('[Cache] Rebuilding hash cache')
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)
print('[Cache] Done')
# write cache to file again for sorting and cleanup of invalid cache entries
save_cache_to_file()

4
modules/hash_cache.py
View File

@ -30,9 +30,8 @@ def load_cache_from_file():
print(f'[Cache] Skipping invalid cache entry: {filepath}')
continue
hash_cache[filepath] = hash_value
print(f'[Cache] Warmed cache from file')
except Exception as e:
print(f'[Cache] Warming failed: {e}')
print(f'[Cache] Loading failed: {e}')
def save_cache_to_file(filename=None, hash_value=None):
@ -50,6 +49,5 @@ def save_cache_to_file(filename=None, hash_value=None):
for filepath, hash_value in items:
json.dump({filepath: hash_value}, fp)
fp.write('\n')
print(f'[Cache] Updated cache file')
except Exception as e:
print(f'[Cache] Saving failed: {e}')