diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-04 19:05:40 +0000 |
|---|---|---|
| committer | AUTOMATIC1111 <16777216c@gmail.com> | 2023-08-04 19:05:50 +0000 |
| commit | 45601766409e531d2b4ee512bf1433600f140183 (patch) | |
| tree | 44fc1f32cd7d8a51242d75d76c7b300d69f6d8ae /modules/extra_networks.py | |
| parent | 31a9966b9d76cb9a2dd7c09c47e236fae33836e2 (diff) | |
| download | stable-diffusion-webui-gfx803-45601766409e531d2b4ee512bf1433600f140183.tar.gz stable-diffusion-webui-gfx803-45601766409e531d2b4ee512bf1433600f140183.tar.bz2 stable-diffusion-webui-gfx803-45601766409e531d2b4ee512bf1433600f140183.zip | |
added VAE selection to checkpoint user metadata
Diffstat (limited to 'modules/extra_networks.py')
| -rw-r--r-- | modules/extra_networks.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/modules/extra_networks.py b/modules/extra_networks.py index 6ae07e91..fa28ac75 100644 --- a/modules/extra_networks.py +++ b/modules/extra_networks.py @@ -1,3 +1,5 @@ +import json
+import os
import re
from collections import defaultdict
@@ -177,3 +179,20 @@ def parse_prompts(prompts): return res, extra_data
+
+def get_user_metadata(filename):
+ if filename is None:
+ return {}
+
+ basename, ext = os.path.splitext(filename)
+ metadata_filename = basename + '.json'
+
+ metadata = {}
+ try:
+ if os.path.isfile(metadata_filename):
+ with open(metadata_filename, "r", encoding="utf8") as file:
+ metadata = json.load(file)
+ except Exception as e:
+ errors.display(e, f"reading extra network user metadata from {metadata_filename}")
+
+ return metadata
|