update init_image to take "None"

2 years ago · 458a925911
parent d84072d922
commit 458a925911
2 changed files with 2 additions and 2 deletions
--- a/Disco_Diffusion.ipynb
+++ b/Disco_Diffusion.ipynb
@ -1068,7 +1068,7 @@
        "      \n",
        "      # Inits if not video frames\n",
        "      if args.animation_mode != \"Video Input\":\n",
-        "        if args.init_image == '':\n",
+        "        if args.init_image == '' or args.init_image.lower() == 'none':\n",
        "          init_image = None\n",
        "        else:\n",
        "          init_image = args.init_image\n",
--- a/disco.py
+++ b/disco.py
@ -1031,7 +1031,7 @@ def do_run():
      
      # Inits if not video frames
      if args.animation_mode != "Video Input":
-        if args.init_image == '':
+        if args.init_image == '' or args.init_image.lower() == 'none':
          init_image = None
        else:
          init_image = args.init_image