added VAE selection to checkpoint user metadata

This commit is contained in:
AUTOMATIC1111
2023-08-04 22:05:40 +03:00
parent 31a9966b9d
commit 4560176640
5 changed files with 96 additions and 12 deletions

View File

@@ -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