From 2cfd10cc856c58442a8fd8119b1aff1d6dd5e855 Mon Sep 17 00:00:00 2001
From: alpertunga-bile <kingdom19online@hotmail.com>
Date: Sat, 12 Aug 2023 14:49:23 +0300
Subject: [PATCH] Model name is added to log output

---
 prompt_generator.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/prompt_generator.py b/prompt_generator.py
index 9efe512..7a98174 100644
--- a/prompt_generator.py
+++ b/prompt_generator.py
@@ -100,6 +100,7 @@ class PromptGenerator:
 
     def log_outputs(
         self,
+        model_name: str,
         prompt: str,
         generated_text: str,
         self_recursive: str,
@@ -119,6 +120,7 @@ class PromptGenerator:
         with open(log_filename, "a") as file:
             file.write(f"{'#'*200}\n")
             file.write(f"Date & Time           : {datetime.now()}\n")
+            file.write(f"Model                 : {model_name}\n")
             file.write(f"Prompt                : {prompt}\n")
             file.write(f"Generated Prompt      : {generated_text}\n")
             file.write(f"cfg                   : {gen_settings.guidance_scale}\n")
@@ -209,6 +211,7 @@ class PromptGenerator:
         )
 
         self.log_outputs(
+            model_name,
             prompt,
             generated_text,
             self_recursive,