Merge pull request #59 from MSFTserver/main

change wierd lower() to a list
2 years ago · af10d0d737
parent 203bd1d199 50e54c2082
commit af10d0d737
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 == '' or args.init_image.lower() == 'none':\n",
+        "        if args.init_image in ['','none', 'None', '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 == '' or args.init_image.lower() == 'none':
+        if args.init_image in ['','none', 'None', 'NONE']:
          init_image = None
        else:
          init_image = args.init_image