From 370ca9e8cdaf95759135819d087d07aaec253afd Mon Sep 17 00:00:00 2001
From: orenwang <371248882@qq.com>
Date: Mon, 13 Mar 2023 14:32:22 +0800
Subject: [PATCH] fix exception on training model in diffusers format

---
 library/train_util.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/library/train_util.py b/library/train_util.py
index 718fe36d..19a21469 100644
--- a/library/train_util.py
+++ b/library/train_util.py
@@ -1183,18 +1183,25 @@ def model_hash(filename):
       return m.hexdigest()[0:8]
   except FileNotFoundError:
     return 'NOFILE'
+  except IsADirectoryError:
+    return 'IsADirectory'
 
 
 def calculate_sha256(filename):
   """New model hash used by stable-diffusion-webui"""
-  hash_sha256 = hashlib.sha256()
-  blksize = 1024 * 1024
+  try:
+    hash_sha256 = hashlib.sha256()
+    blksize = 1024 * 1024
 
-  with open(filename, "rb") as f:
-    for chunk in iter(lambda: f.read(blksize), b""):
-      hash_sha256.update(chunk)
+    with open(filename, "rb") as f:
+      for chunk in iter(lambda: f.read(blksize), b""):
+        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):