diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-27 12:47:33 +0000 |
| commit | 339b5315700a469f4a9f0d5afc08ca2aca60c579 (patch) | |
| tree | 154d6ed405883ca47e23d67e6d3f7699ff4e23e3 /modules/shared_items.py | |
| parent | a6e653be26cc05f4438145fa0082816e9fbbf5fc (diff) | |
| download | stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.gz stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.tar.bz2 stable-diffusion-webui-gfx803-339b5315700a469f4a9f0d5afc08ca2aca60c579.zip | |
custom unet support
Diffstat (limited to 'modules/shared_items.py')
| -rw-r--r-- | modules/shared_items.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/modules/shared_items.py b/modules/shared_items.py index 2a8713c8..7f306a06 100644 --- a/modules/shared_items.py +++ b/modules/shared_items.py @@ -29,3 +29,14 @@ def cross_attention_optimizations(): return ["Automatic"] + [x.title() for x in modules.sd_hijack.optimizers] + ["None"]
+def sd_unet_items():
+ import modules.sd_unet
+
+ return ["Automatic"] + [x.label for x in modules.sd_unet.unet_options] + ["None"]
+
+
+def refresh_unet_list():
+ import modules.sd_unet
+
+ modules.sd_unet.list_unets()
+
|