fix exception on training model in diffusers format

This commit is contained in:
orenwang
2023-03-13 14:32:22 +08:00
parent 7c1cf7f4ea
commit 370ca9e8cd

View File

@@ -1183,18 +1183,25 @@ def model_hash(filename):
return m.hexdigest()[0:8] return m.hexdigest()[0:8]
except FileNotFoundError: except FileNotFoundError:
return 'NOFILE' return 'NOFILE'
except IsADirectoryError:
return 'IsADirectory'
def calculate_sha256(filename): def calculate_sha256(filename):
"""New model hash used by stable-diffusion-webui""" """New model hash used by stable-diffusion-webui"""
hash_sha256 = hashlib.sha256() try:
blksize = 1024 * 1024 hash_sha256 = hashlib.sha256()
blksize = 1024 * 1024
with open(filename, "rb") as f: with open(filename, "rb") as f:
for chunk in iter(lambda: f.read(blksize), b""): for chunk in iter(lambda: f.read(blksize), b""):
hash_sha256.update(chunk) hash_sha256.update(chunk)
return hash_sha256.hexdigest() return hash_sha256.hexdigest()
except FileNotFoundError:
return 'NOFILE'
except IsADirectoryError:
return 'IsADirectory'
def precalculate_safetensors_hashes(tensors, metadata): def precalculate_safetensors_hashes(tensors, metadata):