diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 08:36:07 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-21 08:36:07 +0300 |
| commit | 40ff6db5325fc34ad4fa35e80cb1e7768d9f7e75 (patch) | |
| tree | 75230fd3b1c9c53593aca65d7260b62b6df2d82b /modules/ui_extra_networks_hypernets.py | |
| parent | e33cace2c2074ef342d027c1f31ffc4b3c3e877e (diff) | |
extra networks UI
rework of hypernets: rather than via settings, hypernets are added directly to prompt as <hypernet:name:weight>
Diffstat (limited to 'modules/ui_extra_networks_hypernets.py')
| -rw-r--r-- | modules/ui_extra_networks_hypernets.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/modules/ui_extra_networks_hypernets.py b/modules/ui_extra_networks_hypernets.py new file mode 100644 index 00000000..312dbaf0 --- /dev/null +++ b/modules/ui_extra_networks_hypernets.py @@ -0,0 +1,34 @@ +import os
+
+from modules import shared, ui_extra_networks
+
+
+class ExtraNetworksPageHypernetworks(ui_extra_networks.ExtraNetworksPage):
+ def __init__(self):
+ super().__init__('Hypernetworks')
+
+ def refresh(self):
+ shared.reload_hypernetworks()
+
+ def list_items(self):
+ for name, path in shared.hypernetworks.items():
+ path, ext = os.path.splitext(path)
+ previews = [path + ".png", path + ".preview.png"]
+
+ preview = None
+ for file in previews:
+ if os.path.isfile(file):
+ preview = "./file=" + file.replace('\\', '/') + "?mtime=" + str(os.path.getmtime(file))
+ break
+
+ yield {
+ "name": name,
+ "filename": path,
+ "preview": preview,
+ "prompt": f"<hypernet:{name}:1.0>",
+ "local_preview": path + ".png",
+ }
+
+ def allowed_directories_for_previews(self):
+ return [shared.cmd_opts.hypernetwork_dir]
+
|