img history

ImageState.py

@@ -124,8 +124,9 @@ class ImageState:
             new_latent, _, _ = self.vqgan.quantize(new_latent.to(self.device))
         image = self._decode_latent_to_pil(new_latent)
         img_dir = self.state_id
-        if not os.path.exists(img_dir):
+        if not os.path.exists("img_history"):
             os.mkdir("./img_history")
+        if not os.path.exists(img_dir):
             os.mkdir(img_dir)
         image.save(f"{img_dir}/img_{num:06}.png")
         num += 1

animation.py

@@ -3,8 +3,9 @@ import glob
 import os
 
 def clear_img_dir(img_dir):
-    if not os.path.exists(img_dir):
+    if not os.path.exists("img_history"):
         os.mkdir("img_history")
+    if not os.path.exists(img_dir):
         os.mkdir(img_dir)
     for filename in glob.glob(img_dir+"/*"):
         os.remove(filename)