Commit ce68ab8d authored by AUTOMATIC's avatar AUTOMATIC
Browse files

remove underscores from function names in #8366

remove LRU from #8366 because I don't know why it's there
parent c239b3d7
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -18,8 +18,8 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
            yield {
                "name": name,
                "filename": path,
                "preview": self._find_preview(path),
                "description": self._find_description(path),
                "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(">"),
                "local_preview": path + ".png",
+3 −5
Original line number Diff line number Diff line
import glob
import os.path
import urllib.parse
from functools import lru_cache
from pathlib import Path
from typing import Optional

from modules import shared
import gradio as gr
@@ -140,17 +138,17 @@ class ExtraNetworksPage:

        return self.card_page.format(**args)

    def _find_preview(self, path: str) -> Optional[str]:
    def find_preview(self, path):
        """
        Find a preview PNG for a given path (without extension) and call link_preview on it.
        """
        for file in [path + ".png", path + ".preview.png"]:
            if os.path.isfile(file):
                return self.link_preview(file)

        return None

    @lru_cache(maxsize=512)
    def _find_description(self, path: str) -> Optional[str]:
    def find_description(self, path):
        """
        Find and read a description file for a given path (without extension).
        """
+2 −2
Original line number Diff line number Diff line
@@ -19,8 +19,8 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
            yield {
                "name": checkpoint.name_for_extra,
                "filename": path,
                "preview": self._find_preview(path),
                "description": self._find_description(path),
                "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({json.dumps(name)})""") + '"',
                "local_preview": path + ".png",
+2 −2
Original line number Diff line number Diff line
@@ -18,8 +18,8 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
            yield {
                "name": name,
                "filename": path,
                "preview": self._find_preview(path),
                "description": self._find_description(path),
                "preview": self.find_preview(path),
                "description": self.find_description(path),
                "search_term": self.search_terms_from_path(path),
                "prompt": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
                "local_preview": path + ".png",
+2 −2
Original line number Diff line number Diff line
@@ -18,8 +18,8 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
            yield {
                "name": embedding.name,
                "filename": embedding.filename,
                "preview": self._find_preview(path),
                "description": self._find_description(path),
                "preview": self.find_preview(path),
                "description": self.find_description(path),
                "search_term": self.search_terms_from_path(embedding.filename),
                "prompt": json.dumps(embedding.name),
                "local_preview": path + ".preview.png",