move save params to after generate

This commit is contained in:
Vladimir Mandic
2023-11-11 07:59:48 -05:00
parent b5c0587372
commit 9c7e1291a7
4 changed files with 7 additions and 10 deletions
+5 -1
View File
@@ -4,7 +4,7 @@ from collections import namedtuple
from pathlib import Path
import gradio as gr
from PIL import Image, PngImagePlugin
from modules import shared, errors
from modules import shared, errors, paths
Savedfile = namedtuple("Savedfile", ["name"])
@@ -69,6 +69,10 @@ def pil_to_temp_file(self, img: Image, dir: str, format="png") -> str: # pylint:
name = tmp.name
img.save(name, pnginfo=(metadata if use_metadata else None))
shared.log.debug(f'Saving temp: image="{name}"')
params = ', '.join([f'{k}: {v}' for k, v in img.info.items()])
params = params[12:] if params.startswith('parameters: ') else params
with open(os.path.join(paths.data_path, "params.txt"), "w", encoding="utf8") as file:
file.write(params)
return name