hoelee/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-08-09 05:39:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Include the model name (or the SHA256 of the file) in the metadata #271

Browse Source
This commit is contained in:
AUTOMATIC
2022-09-12 20:47:46 +03:00
parent 35a4649c9e
commit 3de44fc580
4 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules/images.py
View File

@@ -9,7 +9,7 @@ from fonts.ttf import Roboto
import string
import modules.shared
from modules import sd_samplers
from modules import sd_samplers, shared
from modules.shared import opts
LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.LANCZOS)
@@ -278,6 +278,8 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i
file_decoration = file_decoration.replace("[height]", str(p.height))
file_decoration = file_decoration.replace("[sampler]", sd_samplers.samplers[p.sampler_index].name)
file_decoration = file_decoration.replace("[model_hash]", shared.sd_model_hash)
if extension == 'png' and opts.enable_pnginfo and info is not None:
pnginfo = PngImagePlugin.PngInfo()