Merge branch 'dev' into npu_support

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)},