From 86867e153f4449167e3489323df35cf04f1fffa0 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Fri, 9 Sep 2022 23:16:02 +0300
Subject: support for prompt styles fix broken prompt matrix

---
 modules/shared.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'modules/shared.py')

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']
-- 
cgit v1.2.1