'init_image' -> 'image'

yyahav · Mar 22, 2023 · c949b9f · c949b9f
1 parent a07ab84
commit c949b9f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Video_Killed_The_Radio_Star_Defusion.ipynb b/Video_Killed_The_Radio_Star_Defusion.ipynb
@@ -715,6 +715,11 @@
         "\n",
         "    # I give up.\n",
         "    def get_image_for_prompt(*args, **kargs):\n",
+        "        # ugly hotfix. todo: unify function signatures...\n",
+        "        if 'init_image' in kargs:\n",
+        "            img = kargs.pop('init_image')\n",
+        "            if img:\n",
+        "                kargs['image'] = img\n",
         "        result = hf_helper.get_image_for_prompt(*args, **kargs)\n",
         "        return result.images\n",
         "\n",