From c85d440db9af65eaab985ace6bf6a94f8b4c7e03 Mon Sep 17 00:00:00 2001 From: Russell <42530666+hardreddata@users.noreply.github.com> Date: Mon, 21 Apr 2025 05:52:31 +1000 Subject: [PATCH 1/2] Add prompt metadata to png exports --- demo_gradio.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/demo_gradio.py b/demo_gradio.py index c59ed58a..c3346231 100644 --- a/demo_gradio.py +++ b/demo_gradio.py @@ -14,6 +14,7 @@ import math from PIL import Image +from PIL.PngImagePlugin import PngInfo from diffusers import AutoencoderKLHunyuanVideo from transformers import LlamaModel, CLIPTextModel, LlamaTokenizerFast, CLIPTokenizer from diffusers_helper.hunyuan import encode_prompt_conds, vae_decode, vae_encode, vae_decode_fake @@ -141,7 +142,10 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind height, width = find_nearest_bucket(H, W, resolution=640) input_image_np = resize_and_center_crop(input_image, target_width=width, target_height=height) - Image.fromarray(input_image_np).save(os.path.join(outputs_folder, f'{job_id}.png')) + metadata = PngInfo() + metadata.add_text("prompt", prompt) + + Image.fromarray(input_image_np).save(os.path.join(outputs_folder, f'{job_id}.png'), pnginfo=metadata) input_image_pt = torch.from_numpy(input_image_np).float() / 127.5 - 1 input_image_pt = input_image_pt.permute(2, 0, 1)[None, :, None] From cf455daf7bab9d240007b46a3864ece779a6ddab Mon Sep 17 00:00:00 2001 From: Russell <42530666+hardreddata@users.noreply.github.com> Date: Mon, 21 Apr 2025 08:08:36 +1000 Subject: [PATCH 2/2] Add prompt and seed metadata to png --- demo_gradio.py | 1 + 1 file changed, 1 insertion(+) diff --git a/demo_gradio.py b/demo_gradio.py index c3346231..ae028749 100644 --- a/demo_gradio.py +++ b/demo_gradio.py @@ -144,6 +144,7 @@ def worker(input_image, prompt, n_prompt, seed, total_second_length, latent_wind metadata = PngInfo() metadata.add_text("prompt", prompt) + metadata.add_text("seed", str(seed)) Image.fromarray(input_image_np).save(os.path.join(outputs_folder, f'{job_id}.png'), pnginfo=metadata)