add no-metadata

This commit is contained in:
Vladimir Mandic
2023-09-10 08:32:53 -04:00
parent 4898c0ffa7
commit 60df52f636
7 changed files with 20 additions and 24 deletions
+1
View File
@@ -36,6 +36,7 @@ group.add_argument("--tls-certfile", type=str, help="Enable TLS and specify cert
group.add_argument("--tls-selfsign", action="store_true", help="Enable TLS with self-signed certificates, default: %(default)s", default=None)
group.add_argument("--server-name", type=str, help="Sets hostname of server, default: %(default)s", default=None)
group.add_argument("--no-hashing", action='store_true', help="Disable hashing of checkpoints, default: %(default)s", default=False)
group.add_argument("--no-metadata", action='store_true', help="Disable reading of metadata from models, default: %(default)s", default=False)
group.add_argument("--no-download", action='store_true', help="Disable download of default model, default: %(default)s", default=False)
group.add_argument("--profile", action='store_true', help="Run profiler, default: %(default)s")
group.add_argument("--disable-queue", action='store_true', help="Disable queues, default: %(default)s")
+2
View File
@@ -321,6 +321,8 @@ def read_metadata_from_safetensors(filename):
if res is not None:
return res
res = {}
if shared.cmd_opts.no_metadata:
return {}
try:
t0 = time.time()
with open(filename, mode="rb") as file: