Merge pull request #14814 from AUTOMATIC1111/catch-load-style.csv-error
catch load style.csv error
This commit is contained in:
commit
b594f518b7
|
@ -1,4 +1,5 @@
|
|||
from pathlib import Path
|
||||
from modules import errors
|
||||
import csv
|
||||
import os
|
||||
import typing
|
||||
|
@ -128,6 +129,7 @@ class StyleDatabase:
|
|||
self.load_from_csv(styles_file)
|
||||
|
||||
def load_from_csv(self, path: str | Path):
|
||||
try:
|
||||
with open(path, "r", encoding="utf-8-sig", newline="") as file:
|
||||
reader = csv.DictReader(file, skipinitialspace=True)
|
||||
for row in reader:
|
||||
|
@ -141,6 +143,8 @@ class StyleDatabase:
|
|||
self.styles[row["name"]] = PromptStyle(
|
||||
row["name"], prompt, negative_prompt, str(path)
|
||||
)
|
||||
except Exception:
|
||||
errors.report(f'Error loading styles from {path}: ', exc_info=True)
|
||||
|
||||
def get_style_paths(self) -> set:
|
||||
"""Returns a set of all distinct paths of files that styles are loaded from."""
|
||||
|
|
Loading…
Reference in New Issue