add newlines to make params compatible
This commit is contained in:
parent
f142f83ad5
commit
14e920c6e1
|
@ -72,7 +72,7 @@ def str_params(
|
||||||
size: Size,
|
size: Size,
|
||||||
) -> str:
|
) -> str:
|
||||||
return (
|
return (
|
||||||
f"{params.input_prompt}. Negative prompt: {params.input_negative_prompt}."
|
f"{params.input_prompt}.\nNegative prompt: {params.input_negative_prompt}.\n"
|
||||||
f"Steps: {params.steps}, Sampler: {params.scheduler}, CFG scale: {params.cfg}, "
|
f"Steps: {params.steps}, Sampler: {params.scheduler}, CFG scale: {params.cfg}, "
|
||||||
f"Seed: {params.seed}, Size: {size.width}x{size.height}, Model hash: TODO, Model: {params.model}, "
|
f"Seed: {params.seed}, Size: {size.width}x{size.height}, Model hash: TODO, Model: {params.model}, "
|
||||||
f"Version: TODO, Tool: onnx-web"
|
f"Version: TODO, Tool: onnx-web"
|
||||||
|
@ -131,11 +131,9 @@ def save_image(
|
||||||
exif = PngImagePlugin.PngInfo()
|
exif = PngImagePlugin.PngInfo()
|
||||||
|
|
||||||
if params is not None:
|
if params is not None:
|
||||||
exif.add_text("Make", "onnx-web")
|
exif.add_text("make", "onnx-web")
|
||||||
exif.add_text("Model", "TODO: server.version")
|
|
||||||
exif.add_text("Parameters", str_params(params, size))
|
|
||||||
exif.add_text(
|
exif.add_text(
|
||||||
"JSON Parameters",
|
"maker note",
|
||||||
dumps(
|
dumps(
|
||||||
json_params(
|
json_params(
|
||||||
[output],
|
[output],
|
||||||
|
@ -147,15 +145,14 @@ def save_image(
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
exif.add_text("model", "TODO: server.version")
|
||||||
|
exif.add_text("parameters", str_params(params, size))
|
||||||
|
|
||||||
image.save(path, format=server.image_format, pnginfo=exif)
|
image.save(path, format=server.image_format, pnginfo=exif)
|
||||||
else:
|
else:
|
||||||
exif = dump(
|
exif = dump(
|
||||||
{
|
{
|
||||||
"0th": {
|
"0th": {
|
||||||
ExifIFD.UserComment: UserComment.dump(
|
|
||||||
str_params(params, size), encoding="unicode"
|
|
||||||
),
|
|
||||||
ExifIFD.MakerNote: UserComment.dump(
|
ExifIFD.MakerNote: UserComment.dump(
|
||||||
dumps(
|
dumps(
|
||||||
json_params(
|
json_params(
|
||||||
|
@ -169,6 +166,9 @@ def save_image(
|
||||||
),
|
),
|
||||||
encoding="unicode",
|
encoding="unicode",
|
||||||
),
|
),
|
||||||
|
ExifIFD.UserComment: UserComment.dump(
|
||||||
|
str_params(params, size), encoding="unicode"
|
||||||
|
),
|
||||||
ImageIFD.Make: "onnx-web",
|
ImageIFD.Make: "onnx-web",
|
||||||
ImageIFD.Model: "TODO: server.version",
|
ImageIFD.Model: "TODO: server.version",
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue