aboutsummaryrefslogtreecommitdiff
path: root/modules/shared.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-06-27 09:19:04 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-06-27 09:19:04 +0300
commit24129368f1b732be25ef486edb2cf5a6ace66737 (patch)
treee50fa0b140810eeb2698df5bdd49f2acdcc7d6d7 /modules/shared.py
parent14196548c55dfe4775c96bdb939ce1a150933393 (diff)
send tensors to the correct device when loading from safetensors file with memmap disabled for #11260
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/shared.py b/modules/shared.py
index 4743a428..203ee1b9 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -376,7 +376,7 @@ options_templates.update(options_section(('system', "System"), {
"multiple_tqdm": OptionInfo(True, "Add a second progress bar to the console that shows progress for an entire job."),
"print_hypernet_extra": OptionInfo(False, "Print extra hypernetwork information to console."),
"list_hidden_files": OptionInfo(True, "Load models/files in hidden directories").info("directory is hidden if its name starts with \".\""),
- "disable_mmap_load_safetensors": OptionInfo(False, "Disable memmapping for loading .safetensors files (fixes very slow loading speed in some cases)."),
+ "disable_mmap_load_safetensors": OptionInfo(False, "Disable memmapping for loading .safetensors files.").info("fixes very slow loading speed in some cases"),
}))
options_templates.update(options_section(('training', "Training"), {