diff options
| author | unknown <mcgpapu@gmail.com> | 2023-01-28 09:40:51 +0000 |
|---|---|---|
| committer | unknown <mcgpapu@gmail.com> | 2023-01-28 09:40:51 +0000 |
| commit | e79b7db4b47a33889551b9266ee3277879d4f560 (patch) | |
| tree | 1c1944204e58e254bfea22ae44edccdbb54e6b3c /modules/ui_extra_networks_hypernets.py | |
| parent | b921a52071cf2a5e551c31a6073af6eaebbf7847 (diff) | |
| parent | e8a41df49fadd2cf9f23b1f02d75a4947bec5646 (diff) | |
| download | stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.tar.gz stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.tar.bz2 stable-diffusion-webui-gfx803-e79b7db4b47a33889551b9266ee3277879d4f560.zip | |
Merge branch 'master' of github.com:AUTOMATIC1111/stable-diffusion-webui into gamepad
Diffstat (limited to 'modules/ui_extra_networks_hypernets.py')
| -rw-r--r-- | modules/ui_extra_networks_hypernets.py | 35 |
1 files changed, 35 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..65d000cf --- /dev/null +++ b/modules/ui_extra_networks_hypernets.py @@ -0,0 +1,35 @@ +import json
+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": json.dumps(f"<hypernet:{name}:") + " + opts.extra_networks_default_multiplier + " + json.dumps(">"),
+ "local_preview": path + ".png",
+ }
+
+ def allowed_directories_for_previews(self):
+ return [shared.cmd_opts.hypernetwork_dir]
+
|