Add model hash switch to saver

2023-09-23 23:31:39 +08:00
parent 08c6495057
commit 497bfc3b89
1 changed files with 8 additions and 3 deletions
@@ -270,6 +270,11 @@ class SDPromptSaver:
            i = 255.0 * image.cpu().numpy()
            img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
            metadata = None
+            model_hash = (
+                f"Model hash: {self.calculate_model_hash(model_name)}, "
+                if calculate_model_hash
+                else ""
+            )
            comment = (
                f"{positive}\n"
                f"Negative prompt: {negative}\n"
@@ -278,9 +283,9 @@ class SDPromptSaver:
                f"CFG scale: {cfg}, "
                f"Seed: {seed}, "
                f"Size: {img.width if width==0 else width}x{img.height if height==0 else height}, "
-                f"Model hash: {self.calculate_model_hash(model_name)}, "
-                if calculate_model_hash
-                else "" f"Model: {Path(model_name).stem}, " f"Version: ComfyUI"
+                f"{model_hash}"
+                f"Model: {Path(model_name).stem}, "
+                f"Version: ComfyUI"
            )
            file = Path(full_output_folder) / f"{filename}_{counter:05}_.{extension}"
            if extension == "png":