Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

blackify

Browse Source
This commit is contained in:
Lincoln Stein
2023-09-19 00:54:14 -04:00
parent 0c8849155e
commit 73bc088fa7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/model_manager/hash.py
View File

@ -55,7 +55,7 @@ class FastModelHash(object):
for root, dirs, files in os.walk(model_location):
for file in files:
# only tally tensor files
if not file.endswith(('.ckpt', '.safetensors', '.bin', '.pt', '.pth')):
if not file.endswith((".ckpt", ".safetensors", ".bin", ".pt", ".pth")):
continue
path = Path(root) / file
fast_hash = cls._hash_file(path)