VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix exception on training model in diffusers format

Browse Source
This commit is contained in:
orenwang
2023-03-13 14:32:22 +08:00
parent 7c1cf7f4ea
commit 370ca9e8cd
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
library/train_util.py
View File

@@ -1183,10 +1183,13 @@ 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"""
try:
hash_sha256 = hashlib.sha256() hash_sha256 = hashlib.sha256()
blksize = 1024 * 1024 blksize = 1024 * 1024
@@ -1195,6 +1198,10 @@ def calculate_sha256(filename):
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):