OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-02-22 22:25:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix old hash digits (#16845)

Browse Source
This commit is contained in:
w-e-w 2025-02-15 23:49:48 +08:00 committed by GitHub
parent 8c7bc08f60
commit 32595360f2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/hashes.py
View File

@ -83,7 +83,7 @@ def addnet_hash_safetensors(b):
return hash_sha256.hexdigest() return hash_sha256.hexdigest()
def partial_hash_from_cache(filename, ignore_cache=False): def partial_hash_from_cache(filename, *, ignore_cache: bool = False, digits: int = 8):
"""old hash that only looks at a small part of the file and is prone to collisions """old hash that only looks at a small part of the file and is prone to collisions
kept for compatibility, don't use this for new things kept for compatibility, don't use this for new things
""" """
@ -95,7 +95,7 @@ def partial_hash_from_cache(filename, ignore_cache=False):
cache_mtime = cache_entry.get("mtime", 0) cache_mtime = cache_entry.get("mtime", 0)
cache_hash = cache_entry.get("hash", None) cache_hash = cache_entry.get("hash", None)
if mtime == cache_mtime and cache_hash and not ignore_cache: if mtime == cache_mtime and cache_hash and not ignore_cache:
return cache_hash return cache_hash[0:digits]
with open(filename, 'rb') as file: with open(filename, 'rb') as file:
m = hashlib.sha256() m = hashlib.sha256()
@ -103,7 +103,7 @@ def partial_hash_from_cache(filename, ignore_cache=False):
m.update(file.read(0x10000)) m.update(file.read(0x10000))
partial_hash = m.hexdigest() partial_hash = m.hexdigest()
hashes[filename] = {'mtime': mtime, 'hash': partial_hash} hashes[filename] = {'mtime': mtime, 'hash': partial_hash}
return partial_hash[0:8] return partial_hash[0:digits]
except FileNotFoundError: except FileNotFoundError:
pass pass