From ebff19b48a029bf1a3e84f64abc568129facc353 Mon Sep 17 00:00:00 2001 From: filipstrand Date: Mon, 23 Sep 2024 20:02:56 +0200 Subject: [PATCH] Use `None` instead of empty string for image metadata --- src/mflux/post_processing/generated_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/mflux/post_processing/generated_image.py b/src/mflux/post_processing/generated_image.py index add347c..251e57b 100644 --- a/src/mflux/post_processing/generated_image.py +++ b/src/mflux/post_processing/generated_image.py @@ -52,8 +52,8 @@ class GeneratedImage: "precision": f"{self.precision}", "quantization": "None" if self.quantization is None else f"{self.quantization} bit", "generation_time": f"{self.generation_time:.2f} seconds", - "lora_paths": ", ".join(self.lora_paths) if self.lora_paths else "", - "lora_scales": ", ".join([f"{scale:.2f}" for scale in self.lora_scales]) if self.lora_scales else "", + "lora_paths": ", ".join(self.lora_paths) if self.lora_paths else "None", + "lora_scales": ", ".join([f"{scale:.2f}" for scale in self.lora_scales]) if self.lora_scales else "None", "prompt": self.prompt, "controlnet_image": "None" if self.controlnet_image is None else self.controlnet_image, "controlnet_strength": "None" if self.controlnet_strength is None else f"{self.controlnet_strength:.2f}",