Unverified Commit 9009e25c authored by catboxanon's avatar catboxanon Committed by GitHub
Browse files

Apply suggestions from code review



Co-authored-by: default avatarAarni Koskela <akx@iki.fi>
parent 4cc0cede
Loading
Loading
Loading
Loading
+2 −1
Original line number Diff line number Diff line
@@ -27,7 +27,8 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
                "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,
                "sort_keys": {**self.get_sort_keys(lora_on_disk.filename), **{'default': index}},
                "sort_keys": {'default': index, **self.get_sort_keys(lora_on_disk.filename)},

            }

    def allowed_directories_for_previews(self):
+5 −3
Original line number Diff line number Diff line
@@ -210,10 +210,12 @@ class ExtraNetworksPage:
        """
        List of default keys used for sorting in the UI.
        """
        pth = Path(path)
        stat = pth.stat()
        return {
            "date_created": int(Path(path).stat().st_ctime or 0),
            "date_modified": int(Path(path).stat().st_mtime or 0),
            "name": Path(path).name.lower(),
            "date_created": int(stat.st_ctime or 0),
            "date_modified": int(stat.st_mtime or 0),
            "name": pth.name.lower(),
        }

    def find_preview(self, path):
+2 −1
Original line number Diff line number Diff line
@@ -24,7 +24,8 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
                "search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),
                "onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"',
                "local_preview": f"{path}.{shared.opts.samples_format}",
                "sort_keys": {**self.get_sort_keys(checkpoint.filename), **{'default': index}},
                "sort_keys": {'default': index, **self.get_sort_keys(checkpoint.filename)},

            }

    def allowed_directories_for_previews(self):
+2 −1
Original line number Diff line number Diff line
@@ -23,7 +23,8 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
                "search_term": self.search_terms_from_path(path),
                "prompt": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
                "local_preview": f"{path}.preview.{shared.opts.samples_format}",
                "sort_keys": {**self.get_sort_keys(path + ext), **{'default': index}},
                "sort_keys": {'default': index, **self.get_sort_keys(path + ext)},

            }

    def allowed_directories_for_previews(self):
+2 −1
Original line number Diff line number Diff line
@@ -23,7 +23,8 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
                "search_term": self.search_terms_from_path(embedding.filename),
                "prompt": json.dumps(embedding.name),
                "local_preview": f"{path}.preview.{shared.opts.samples_format}",
                "sort_keys": {**self.get_sort_keys(embedding.filename), **{'default': index}},
                "sort_keys": {'default': index, **self.get_sort_keys(embedding.filename)},

            }

    def allowed_directories_for_previews(self):