Update Disco_Diffusion.ipynb

2 years ago · fc2720f1f7
parent bba27b59d2
commit fc2720f1f7
1 changed files with 1 additions and 1 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",