calculate hashes for Lora

add lora hashes to infotext
when pasting infotext, use infotext's lora hashes to find local loras for <lora:xxx:1> entries whose hashes match loras the user has

1 files changed, 23 insertions, 6 deletions

diff --git a/modules/hashes.py b/modules/hashes.py
index 032120f4..8b7ea0ac 100644
--- a/modules/hashes.py
+++ b/modules/hashes.py
@@ -46,8 +46,8 @@ def calculate_sha256(filename):
     return hash_sha256.hexdigest()

 

 

-def sha256_from_cache(filename, title):

-    hashes = cache("hashes")

+def sha256_from_cache(filename, title, use_addnet_hash=False):

+    hashes = cache("hashes-addnet") if use_addnet_hash else cache("hashes")

     ondisk_mtime = os.path.getmtime(filename)

 

     if title not in hashes:

@@ -62,10 +62,10 @@ def sha256_from_cache(filename, title):
     return cached_sha256

 

 

-def sha256(filename, title):

-    hashes = cache("hashes")

+def sha256(filename, title, use_addnet_hash=False):

+    hashes = cache("hashes-addnet") if use_addnet_hash else cache("hashes")

 

-    sha256_value = sha256_from_cache(filename, title)

+    sha256_value = sha256_from_cache(filename, title, use_addnet_hash)

     if sha256_value is not None:

         return sha256_value

 

@@ -73,7 +73,11 @@ def sha256(filename, title):
         return None

 

     print(f"Calculating sha256 for {filename}: ", end='')

-    sha256_value = calculate_sha256(filename)

+    if use_addnet_hash:

+        with open(filename, "rb") as file:

+            sha256_value = addnet_hash_safetensors(file)

+    else:

+        sha256_value = calculate_sha256(filename)

     print(f"{sha256_value}")

 

     hashes[title] = {

@@ -86,6 +90,19 @@ def sha256(filename, title):
     return sha256_value

 

 

+def addnet_hash_safetensors(b):

+    """kohya-ss hash for safetensors from https://github.com/kohya-ss/sd-scripts/blob/main/library/train_util.py"""

+    hash_sha256 = hashlib.sha256()

+    blksize = 1024 * 1024

 

+    b.seek(0)

+    header = b.read(8)

+    n = int.from_bytes(header, "little")

 

+    offset = n + 8

+    b.seek(offset)

+    for chunk in iter(lambda: b.read(blksize), b""):

+        hash_sha256.update(chunk)

+

+    return hash_sha256.hexdigest()