make a dropdown for prompt template selection

1 files changed, 3 insertions, 0 deletions

diff --git a/webui.py b/webui.py
index 8737e593..47d372c7 100644
--- a/webui.py
+++ b/webui.py
@@ -33,6 +33,7 @@ import modules.sd_models
 import modules.sd_vae

 import modules.txt2img

 import modules.script_callbacks

+import modules.textual_inversion.textual_inversion

 

 import modules.ui

 from modules import modelloader

@@ -67,6 +68,8 @@ def initialize():
 

     modules.sd_vae.refresh_vae_list()

 

+    modules.textual_inversion.textual_inversion.list_textual_inversion_templates()

+

     try:

         modules.sd_models.load_model()

     except Exception as e: