fix styles.csv filename

AUTOMATIC1111 · Dec 10, 2023 · ee1463d · ee1463d
1 parent 3e8776a
commit ee1463d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/modules/styles.py b/modules/styles.py
@@ -98,10 +98,8 @@ def __init__(self, path: str):
         self.path = path
 
         folder, file = os.path.split(self.path)
-        self.default_file = file.split("*")[0] + ".csv"
-        if self.default_file == ".csv":
-            self.default_file = "styles.csv"
-        self.default_path = os.path.join(folder, self.default_file)
+        filename, _, ext = file.partition('*')
+        self.default_path = os.path.join(folder, filename + ext)
 
         self.prompt_fields = [field for field in PromptStyle._fields if field != "path"]