StableDiffusion/WebUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: Aarni Koskela <akx@iki.fi>
This commit is contained in:
catboxanon 2023-06-02 16:12:24 -04:00 committed by GitHub
parent 4cc0cede6d
commit 9009e25cb1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
extensions-builtin/Lora/ui_extra_networks_lora.py
View File

@ -27,7 +27,8 @@ class ExtraNetworksPageLora(ui_extra_networks.ExtraNetworksPage):
"prompt": json.dumps(f"<lora:{alias}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"), "prompt": json.dumps(f"<lora:{alias}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
"local_preview": f"{path}.{shared.opts.samples_format}", "local_preview": f"{path}.{shared.opts.samples_format}",
"metadata": json.dumps(lora_on_disk.metadata, indent=4) if lora_on_disk.metadata else None, "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): def allowed_directories_for_previews(self):

8
modules/ui_extra_networks.py
View File

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

3
modules/ui_extra_networks_checkpoints.py
View File

@ -24,7 +24,8 @@ class ExtraNetworksPageCheckpoints(ui_extra_networks.ExtraNetworksPage):
"search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""), "search_term": self.search_terms_from_path(checkpoint.filename) + " " + (checkpoint.sha256 or ""),
"onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"', "onclick": '"' + html.escape(f"""return selectCheckpoint({json.dumps(name)})""") + '"',
"local_preview": f"{path}.{shared.opts.samples_format}", "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): def allowed_directories_for_previews(self):

3
modules/ui_extra_networks_hypernets.py
View File

@ -23,7 +23,8 @@ class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
"search_term": self.search_terms_from_path(path), "search_term": self.search_terms_from_path(path),
"prompt": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"), "prompt": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
"local_preview": f"{path}.preview.{shared.opts.samples_format}", "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): def allowed_directories_for_previews(self):

3
modules/ui_extra_networks_textual_inversion.py
View File

@ -23,7 +23,8 @@ class ExtraNetworksPageTextualInversion(ui_extra_networks.ExtraNetworksPage):
"search_term": self.search_terms_from_path(embedding.filename), "search_term": self.search_terms_from_path(embedding.filename),
"prompt": json.dumps(embedding.name), "prompt": json.dumps(embedding.name),
"local_preview": f"{path}.preview.{shared.opts.samples_format}", "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): def allowed_directories_for_previews(self):