mrq/stable-diffusion-webui
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support NAI exif for PNG Info

Browse Source
This commit is contained in:
Nandaka 2022-11-24 02:39:09 +00:00
parent 828438b4a1
commit 904121fecc
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
modules/extras.py
View File

@ -233,6 +233,20 @@ def run_pnginfo(image):
geninfo = items.get('parameters', geninfo) geninfo = items.get('parameters', geninfo)
# nai prompt
if "Software" in items.keys() and items["Software"] == "NovelAI":
import json
json_info = json.loads(items["Comment"])
geninfo = f'{items["Description"]}\r\nNegative prompt: {json_info["uc"]}\r\n'
sampler = "Euler a"
if json_info["sampler"] == "k_euler_ancestral":
sampler = "Euler a"
elif json_info["sampler"] == "k_euler":
sampler = "Euler"
model_hash = '925997e9' # assuming this is the correct model hash
# not sure with noise and strength parameter
geninfo += f'Steps: {json_info["steps"]}, Sampler: {sampler}, CFG scale: {json_info["scale"]}, Seed: {json_info["seed"]}, Size: {image.width}x{image.height}, Model hash: {model_hash}' # , Denoising strength: {json_info["noise"]}'
info = '' info = ''
for key, text in items.items(): for key, text in items.items():
info += f""" info += f"""