From cf4b85a0a0298038824bc8190e5f1c8c06df7ab8 Mon Sep 17 00:00:00 2001
From: Rhys Yang <receyuki@gmail.com>
Date: Fri, 5 Jan 2024 22:50:22 +0800
Subject: [PATCH] Add temporary storage for model hash

---
 nodes.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index b191335..f70e05d 100644
--- a/nodes.py
+++ b/nodes.py
@@ -240,6 +240,8 @@ class SDPromptReader:
 
 
 class SDPromptSaver:
+    hash_dict = {}
+
     def __init__(self):
         self.output_dir = folder_paths.get_output_directory()
         self.type = "output"
@@ -469,6 +471,9 @@ class SDPromptSaver:
 
     @staticmethod
     def calculate_model_hash(model_name):
+        if hash_value := SDPromptSaver.hash_dict.get(model_name):
+            return hash_value
+
         hash_sha256 = hashlib.sha256()
         blksize = 1024 * 1024
         file_name = folder_paths.get_full_path("checkpoints", model_name)
@@ -477,7 +482,10 @@ class SDPromptSaver:
             for chunk in iter(lambda: f.read(blksize), b""):
                 hash_sha256.update(chunk)
 
-        return hash_sha256.hexdigest()[:10]
+        hash_value = hash_sha256.hexdigest()[:10]
+        SDPromptSaver.hash_dict[model_name] = hash_value
+
+        return hash_value
 
     @staticmethod
     def get_counter(directory: Path):