Update text2img.py

Beyondo · Dec 11, 2022 · c9c644a · c9c644a
1 parent 862126b
commit c9c644a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/text2img.py b/text2img.py
@@ -8,8 +8,8 @@ def process(seed, positive_prompt, negative_prompt, guidance_scale, inference_st
     if save_to_google_drive:
         dir = '/content/gdrive/MyDrive/' + directory
         if not os.path.exists(dir): os.makedirs(dir)
-        imgSavePath = "%s/%d-voidops" % (dir, int(time.mktime(datetime.now().timetuple())))
+        imgSavePath = "%s/%d-voidops" % (dir, int(time.mktime(datetime.datetime.now().timetuple())))
         image.save(imgSavePath + ".png")
-        print("Saved to " + imgSavePath)
         display(image)
+        print("Saved to " + imgSavePath)
         return image