add an option to always refer to lora by filenames (47719935) · Commits · github_fork / Stable Diffusion Webui

extensions-builtin/Lora/lora.py

+6 −0

Original line number	Diff line number	Diff line
		@@ -393,6 +393,8 @@ def lora_MultiheadAttention_load_state_dict(self, args, *kwargs):
		def list_available_loras():
		available_loras.clear()
		available_lora_aliases.clear()
		forbidden_lora_aliases.clear()
		forbidden_lora_aliases.update({"none": 1})

		os.makedirs(shared.cmd_opts.lora_dir, exist_ok=True)

		@@ -406,6 +408,9 @@ def list_available_loras():

		available_loras[name] = entry

		if entry.alias in available_lora_aliases:
		forbidden_lora_aliases[entry.alias.lower()] = 1

		available_lora_aliases[name] = entry
		available_lora_aliases[entry.alias] = entry

		@@ -445,6 +450,7 @@ def infotext_pasted(infotext, params):

		available_loras = {}
		available_lora_aliases = {}
		forbidden_lora_aliases = {}
		loaded_loras = []

		list_available_loras()

+1 −0

Original line number	Diff line number	Diff line
		@@ -54,6 +54,7 @@ script_callbacks.on_infotext_pasted(lora.infotext_pasted)

		shared.options_templates.update(shared.options_section(('extra_networks', "Extra Networks"), {
		"sd_lora": shared.OptionInfo("None", "Add Lora to prompt", gr.Dropdown, lambda: {"choices": ["None"] + [x for x in lora.available_loras]}, refresh=lora.list_available_loras),
		"lora_preferred_name": shared.OptionInfo("Alias from file", "When adding to prompt, refer to lora by", gr.Radio, {"choices": ["Alias from file", "Filename"]}),
		}))

+7 −1

Original line number	Diff line number	Diff line
		@@ -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,
		}