initial support for training textual inversion

1 files changed, 2 insertions, 1 deletions

diff --git a/modules/shared.py b/modules/shared.py
index ac968b2d..ac0bc480 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -78,6 +78,7 @@ class State:
     current_latent = None

     current_image = None

     current_image_sampling_step = 0

+    textinfo = None

 

     def interrupt(self):

         self.interrupted = True

@@ -88,7 +89,7 @@ class State:
         self.current_image_sampling_step = 0

         

     def get_job_timestamp(self):

-        return datetime.datetime.now().strftime("%Y%m%d%H%M%S")

+        return datetime.datetime.now().strftime("%Y%m%d%H%M%S")  # shouldn't this return job_timestamp?

 

 

 state = State()