diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2024-01-22 22:24:06 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-01-22 22:24:06 +0300 |
| commit | 569dc1919c23329ceaec2162f86ca0fafdee39f5 (patch) | |
| tree | d8b6b6591fe83eb8eaa5d0f5e46d7a434b58f8de /modules/ui_extra_networks_checkpoints.py | |
| parent | 8a6a4ad894c2e28bcb6924490396997bfed7f07f (diff) | |
| parent | 26e1cd7ec47c8d234d2ea3f189b1147329c9059c (diff) | |
Merge pull request #14588 from Sj-Si/feature/extra-networks-tree-view
Feature: Extra Networks Tree View
Diffstat (limited to 'modules/ui_extra_networks_checkpoints.py')
| -rw-r--r-- | modules/ui_extra_networks_checkpoints.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/modules/ui_extra_networks_checkpoints.py b/modules/ui_extra_networks_checkpoints.py index 1693e71f..a8c33671 100644 --- a/modules/ui_extra_networks_checkpoints.py +++ b/modules/ui_extra_networks_checkpoints.py @@ -2,7 +2,6 @@ import html import os
from modules import shared, ui_extra_networks, sd_models
-from modules.ui_extra_networks import quote_js
from modules.ui_extra_networks_checkpoints_user_metadata import CheckpointUserMetadataEditor
@@ -21,14 +20,17 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage): return
path, ext = os.path.splitext(checkpoint.filename)
+ search_terms = [self.search_terms_from_path(checkpoint.filename)]
+ if checkpoint.sha256:
+ search_terms.append(checkpoint.sha256)
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 ""),
- "onclick": '"' + html.escape(f"""return selectCheckpoint({quote_js(name)})""") + '"',
+ "search_terms": search_terms,
+ "onclick": html.escape(f"return selectCheckpoint('{name}');"),
"local_preview": f"{path}.{shared.opts.samples_format}",
"metadata": checkpoint.metadata,
"sort_keys": {'default': index, **self.get_sort_keys(checkpoint.filename)},
|