diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-05-13 20:15:37 +0300 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-13 20:15:37 +0300 |
commit | 477199357f4f5f02d62857a0cf432a3ed19e6418 (patch) | |
tree | 75835cbad5302ff658273a06d91a111313deae24 /extensions-builtin/Lora/ui_extra_networks_lora.py | |
parent | 231562ea13e4f697953bdbabd6b76b22a88c587b (diff) |
add an option to always refer to lora by filenames
never refer to lora by an alias if multiple loras have same alias or the alias is called none
Diffstat (limited to 'extensions-builtin/Lora/ui_extra_networks_lora.py')
-rw-r--r-- | extensions-builtin/Lora/ui_extra_networks_lora.py | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/extensions-builtin/Lora/ui_extra_networks_lora.py b/extensions-builtin/Lora/ui_extra_networks_lora.py index a0edbc1e..2050e3fa 100644 --- a/extensions-builtin/Lora/ui_extra_networks_lora.py +++ b/extensions-builtin/Lora/ui_extra_networks_lora.py @@ -15,13 +15,19 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage): def list_items(self):
for name, lora_on_disk in lora.available_loras.items():
path, ext = os.path.splitext(lora_on_disk.filename)
+
+ if shared.opts.lora_preferred_name == "Filename" or lora_on_disk.alias.lower() in lora.forbidden_lora_aliases:
+ alias = name
+ else:
+ alias = lora_on_disk.alias
+
yield {
"name": name,
"filename": path,
"preview": self.find_preview(path),
"description": self.find_description(path),
"search_term": self.search_terms_from_path(lora_on_disk.filename),
- "prompt": json.dumps(f"<lora:{lora_on_disk.alias}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
+ "prompt": json.dumps(f"<lora:{alias}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
"local_preview": f"{path}.{shared.opts.samples_format}",
"metadata": json.dumps(lora_on_disk.metadata, indent=4) if lora_on_disk.metadata else None,
}
|