From fc2720f1f782114b7e7aea1b5696ea1cbb6c1225 Mon Sep 17 00:00:00 2001
From: MSFTserver <MSFTserver@gmail.com>
Date: Thu, 21 Apr 2022 21:14:22 -0700
Subject: [PATCH] Update Disco_Diffusion.ipynb

---
 Disco_Diffusion.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/Disco_Diffusion.ipynb b/Disco_Diffusion.ipynb
index ee4d7c8..5cfa846 100644
--- 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",