if present, use Lora's "ss_output_name" field to refer to it in prompt (2cb3b0be) · Commits · github_fork / Stable Diffusion Webui

extensions-builtin/Lora/extra_networks_lora.py

+1 −0

Original line number	Diff line number	Diff line
		from modules import extra_networks, shared
		import lora


		class ExtraNetworkLora(extra_networks.ExtraNetwork):
		def __init__(self):
		super().__init__('lora')

+10 −3

Original line number	Diff line number	Diff line
		@@ -93,6 +93,7 @@ class LoraOnDisk:
		self.metadata = m

		self.ssmd_cover_images = self.metadata.pop('ssmd_cover_images', None) # those are cover images and they are too big to display in UI as text
		self.alias = self.metadata.get('ss_output_name', self.name)


		class LoraModule:
		@@ -199,11 +200,11 @@ def load_loras(names, multipliers=None):

		loaded_loras.clear()

		loras_on_disk = [available_loras.get(name, None) for name in names]
		loras_on_disk = [available_lora_aliases.get(name, None) for name in names]
		if any([x is None for x in loras_on_disk]):
		list_available_loras()

		loras_on_disk = [available_loras.get(name, None) for name in names]
		loras_on_disk = [available_lora_aliases.get(name, None) for name in names]

		for i, name in enumerate(names):
		lora = already_loaded.get(name, None)
		@@ -343,6 +344,7 @@ def lora_MultiheadAttention_load_state_dict(self, args, *kwargs):

		def list_available_loras():
		available_loras.clear()
		available_lora_aliases.clear()

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

		@@ -356,11 +358,16 @@ def list_available_loras():
		continue

		name = os.path.splitext(os.path.basename(filename))[0]
		entry = LoraOnDisk(name, filename)

		available_loras[name] = LoraOnDisk(name, filename)
		available_loras[name] = entry

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


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

		list_available_loras()

+1 −1

Original line number	Diff line number	Diff line
		@@ -21,7 +21,7 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
		"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:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
		"prompt": json.dumps(f"<lora:{lora_on_disk.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,
		}