From df598c4df5ee8c6c6b12f1cd4c78043c6192cde2 Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Thu, 22 Aug 2024 23:23:14 -0700
Subject: [PATCH] add "Diffusion in Low Bits" to info text

---
 modules/processing.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/modules/processing.py b/modules/processing.py
index 4fbca99c..8b9b46b3 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -766,6 +766,9 @@ def create_infotext(p, all_prompts, all_seeds, all_subseeds, comments=None, iter
         "User": p.user if opts.add_user_name_to_info else None,
     })
 
+    if shared.opts.forge_unet_storage_dtype != 'Automatic':
+        generation_params['Diffusion in Low Bits'] = shared.opts.forge_unet_storage_dtype
+
     if isinstance(shared.opts.forge_additional_modules, list) and len(shared.opts.forge_additional_modules) > 0:
         for i, m in enumerate(shared.opts.forge_additional_modules):
             generation_params[f'Module {i+1}'] = os.path.splitext(os.path.basename(m))[0]