Add FP8 settings into PNG info

2 files changed, 8 insertions, 0 deletions

diff --git a/modules/generation_parameters_copypaste.py b/modules/generation_parameters_copypaste.py
index 4efe53e0..dbffe494 100644
--- a/modules/generation_parameters_copypaste.py
+++ b/modules/generation_parameters_copypaste.py
@@ -314,6 +314,12 @@ Steps: 20, Sampler: Euler a, CFG scale: 7, Seed: 965400086, Size: 512x512, Model
     if "VAE Decoder" not in res:

         res["VAE Decoder"] = "Full"

 

+    if "FP8 weight" not in res:

+        res["FP8 weight"] = "Disable"

+

+    if "Cache FP16 weight for LoRA" not in res and res["FP8 weight"] != "Disable":

+        res["Cache FP16 weight for LoRA"] = False

+

     skip = set(shared.opts.infotext_skip_pasting)

     res = {k: v for k, v in res.items() if k not in skip}

 

diff --git a/modules/processing.py b/modules/processing.py
index bea01ec6..179f2c0f 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -688,6 +688,8 @@ def create_infotext(p, all_prompts, all_seeds, all_subseeds, comments=None, iter
         "Size": f"{p.width}x{p.height}",

         "Model hash": p.sd_model_hash if opts.add_model_hash_to_info else None,

         "Model": p.sd_model_name if opts.add_model_name_to_info else None,

+        "FP8 weight": opts.fp8_storage if devices.fp8 else None,

+        "Cache FP16 weight for LoRA": opts.cache_fp16_weight if devices.fp8 else None,

         "VAE hash": p.sd_vae_hash if opts.add_vae_hash_to_info else None,

         "VAE": p.sd_vae_name if opts.add_vae_name_to_info else None,

         "Variation seed": (None if p.subseed_strength == 0 else (p.all_subseeds[0] if use_main_prompt else all_subseeds[index])),