Support search and display of hashes for all extra network items (7fa5ee54) · Commits · github_fork / Stable Diffusion Webui

extensions-builtin/Lora/ui_extra_networks_lora.py

+2 −1

Original line number	Diff line number	Diff line
		@@ -25,9 +25,10 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
		item = {
		"name": name,
		"filename": lora_on_disk.filename,
		"shorthash": lora_on_disk.shorthash,
		"preview": self.find_preview(path),
		"description": self.find_description(path),
		"search_term": self.search_terms_from_path(lora_on_disk.filename),
		"search_term": self.search_terms_from_path(lora_on_disk.filename) + " " + (lora_on_disk.hash or ""),
		"local_preview": f"{path}.{shared.opts.samples_format}",
		"metadata": lora_on_disk.metadata,
		"sort_keys": {'default': index, **self.get_sort_keys(lora_on_disk.filename)},

+1 −0

Original line number	Diff line number	Diff line
		@@ -19,6 +19,7 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
		return {
		"name": checkpoint.name_for_extra,
		"filename": checkpoint.filename,
		"shorthash": checkpoint.shorthash,
		"preview": self.find_preview(path),
		"description": self.find_description(path),
		"search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),

+5 −1

Original line number	Diff line number	Diff line
		@@ -2,6 +2,7 @@ import os

		from modules import shared, ui_extra_networks
		from modules.ui_extra_networks import quote_js
		from modules.hashes import sha256_from_cache


		class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
		@@ -14,13 +15,16 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
		def create_item(self, name, index=None, enable_filter=True):
		full_path = shared.hypernetworks[name]
		path, ext = os.path.splitext(full_path)
		sha256 = sha256_from_cache(full_path, f'hypernet/{name}')
		shorthash = sha256[0:10] if sha256 else None

		return {
		"name": name,
		"filename": full_path,
		"shorthash": shorthash,
		"preview": self.find_preview(path),
		"description": self.find_description(path),
		"search_term": self.search_terms_from_path(path),
		"search_term": self.search_terms_from_path(path) + " " + (sha256 or ""),
		"prompt": quote_js(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + quote_js(">"),
		"local_preview": f"{path}.preview.{shared.opts.samples_format}",
		"sort_keys": {'default': index, **self.get_sort_keys(path + ext)},

+2 −1

Original line number	Diff line number	Diff line
		@@ -19,9 +19,10 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
		return {
		"name": name,
		"filename": embedding.filename,
		"shorthash": embedding.shorthash,
		"preview": self.find_preview(path),
		"description": self.find_description(path),
		"search_term": self.search_terms_from_path(embedding.filename),
		"search_term": self.search_terms_from_path(embedding.filename) + " " + (embedding.hash or ""),
		"prompt": quote_js(embedding.name),
		"local_preview": f"{path}.preview.{shared.opts.samples_format}",
		"sort_keys": {'default': index, **self.get_sort_keys(embedding.filename)},

+3 −1

Original line number	Diff line number	Diff line
		@@ -93,11 +93,13 @@ class UserMetadataEditor:
		item = self.page.items.get(name, {})
		try:
		filename = item["filename"]
		shorthash = item.get("shorthash", None)

		stats = os.stat(filename)
		params = [
		('Filename: ', os.path.basename(filename)),
		('File size: ', sysinfo.pretty_bytes(stats.st_size)),
		('Hash: ', shorthash),
		('Modified: ', datetime.datetime.fromtimestamp(stats.st_mtime).strftime('%Y-%m-%d %H:%M')),
		]

		@@ -115,7 +117,7 @@ class UserMetadataEditor:
		errors.display(e, f"reading metadata info for {name}")
		params = []

		table = '<table class="file-metadata">' + "".join(f"<tr><th>{name}</th><td>{value}</td></tr>" for name, value in params) + '</table>'
		table = '<table class="file-metadata">' + "".join(f"<tr><th>{name}</th><td>{value}</td></tr>" for name, value in params if value is not None) + '</table>'

		return html.escape(name), user_metadata.get('description', ''), table, self.get_card_html(name), user_metadata.get('notes', '')