Commit 81823407 authored by AUTOMATIC's avatar AUTOMATIC
Browse files

add --no-hashing

parent 30228c67
Loading
Loading
Loading
Loading
+4 −0
Original line number Diff line number Diff line
@@ -4,6 +4,7 @@ import os.path

import filelock

from modules import shared
from modules.paths import data_path


@@ -68,6 +69,9 @@ def sha256(filename, title):
    if sha256_value is not None:
        return sha256_value

    if shared.cmd_opts.no_hashing:
        return None

    print(f"Calculating sha256 for {filename}: ", end='')
    sha256_value = calculate_sha256(filename)
    print(f"{sha256_value}")
+1 −1
Original line number Diff line number Diff line
@@ -307,7 +307,7 @@ class Hypernetwork:
    def shorthash(self):
        sha256 = hashes.sha256(self.filename, f'hypernet/{self.name}')

        return sha256[0:10]
        return sha256[0:10] if sha256 else None


def list_hypernetworks(path):
+3 −0
Original line number Diff line number Diff line
@@ -59,6 +59,9 @@ class CheckpointInfo:

    def calculate_shorthash(self):
        self.sha256 = hashes.sha256(self.filename, "checkpoint/" + self.name)
        if self.sha256 is None:
            return

        self.shorthash = self.sha256[0:10]

        if self.shorthash not in self.ids:
+1 −1
Original line number Diff line number Diff line
@@ -106,7 +106,7 @@ parser.add_argument("--tls-certfile", type=str, help="Partially enables TLS, req
parser.add_argument("--server-name", type=str, help="Sets hostname of server", default=None)
parser.add_argument("--gradio-queue", action='store_true', help="Uses gradio queue; experimental option; breaks restart UI button")
parser.add_argument("--skip-version-check", action='store_true', help="Do not check versions of torch and xformers")

parser.add_argument("--no-hashing", action='store_true', help="disable sha256 hashing of checkpoints to help loading performance", default=False)


script_loading.preload_extensions(extensions.extensions_dir, parser)