add --no-hashing
This commit is contained in:
parent
30228c67ca
commit
81823407d9
@ -4,6 +4,7 @@ import os.path
|
|||||||
|
|
||||||
import filelock
|
import filelock
|
||||||
|
|
||||||
|
from modules import shared
|
||||||
from modules.paths import data_path
|
from modules.paths import data_path
|
||||||
|
|
||||||
|
|
||||||
@ -68,6 +69,9 @@ def sha256(filename, title):
|
|||||||
if sha256_value is not None:
|
if sha256_value is not None:
|
||||||
return sha256_value
|
return sha256_value
|
||||||
|
|
||||||
|
if shared.cmd_opts.no_hashing:
|
||||||
|
return None
|
||||||
|
|
||||||
print(f"Calculating sha256 for {filename}: ", end='')
|
print(f"Calculating sha256 for {filename}: ", end='')
|
||||||
sha256_value = calculate_sha256(filename)
|
sha256_value = calculate_sha256(filename)
|
||||||
print(f"{sha256_value}")
|
print(f"{sha256_value}")
|
||||||
|
@ -307,7 +307,7 @@ class Hypernetwork:
|
|||||||
def shorthash(self):
|
def shorthash(self):
|
||||||
sha256 = hashes.sha256(self.filename, f'hypernet/{self.name}')
|
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):
|
def list_hypernetworks(path):
|
||||||
|
@ -59,6 +59,9 @@ class CheckpointInfo:
|
|||||||
|
|
||||||
def calculate_shorthash(self):
|
def calculate_shorthash(self):
|
||||||
self.sha256 = hashes.sha256(self.filename, "checkpoint/" + self.name)
|
self.sha256 = hashes.sha256(self.filename, "checkpoint/" + self.name)
|
||||||
|
if self.sha256 is None:
|
||||||
|
return
|
||||||
|
|
||||||
self.shorthash = self.sha256[0:10]
|
self.shorthash = self.sha256[0:10]
|
||||||
|
|
||||||
if self.shorthash not in self.ids:
|
if self.shorthash not in self.ids:
|
||||||
|
@ -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("--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("--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("--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)
|
script_loading.preload_extensions(extensions.extensions_dir, parser)
|
||||||
|
Loading…
Reference in New Issue
Block a user