aboutsummaryrefslogtreecommitdiffstats
path: root/modules/shared.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2022-09-09 20:16:02 +0000
committerAUTOMATIC <16777216c@gmail.com>2022-09-09 20:16:02 +0000
commit86867e153f4449167e3489323df35cf04f1fffa0 (patch)
tree735b7311b84136c87ac17472ed464d1d7fb6759f /modules/shared.py
parentd714ea4c41e4e7c0dc4730d9ea137d134691c0a2 (diff)
downloadstable-diffusion-webui-gfx803-86867e153f4449167e3489323df35cf04f1fffa0.tar.gz
stable-diffusion-webui-gfx803-86867e153f4449167e3489323df35cf04f1fffa0.tar.bz2
stable-diffusion-webui-gfx803-86867e153f4449167e3489323df35cf04f1fffa0.zip
support for prompt styles
fix broken prompt matrix
Diffstat (limited to 'modules/shared.py')
-rw-r--r--modules/shared.py6
1 files changed, 4 insertions, 2 deletions
diff --git a/modules/shared.py b/modules/shared.py
index e577332d..5985d09e 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -9,7 +9,7 @@ import tqdm
import modules.artists
from modules.paths import script_path, sd_path
-import modules.codeformer_model
+import modules.styles
config_filename = "config.json"
@@ -75,8 +75,10 @@ state = State()
artist_db = modules.artists.ArtistsDatabase(os.path.join(script_path, 'artists.csv'))
-face_restorers = []
+styles_filename = os.path.join(script_path, 'styles.csv')
+prompt_styles = modules.styles.load_styles(styles_filename)
+face_restorers = []
def find_any_font():
fonts = ['/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf']