Update demo.ipynb

Author: hlky

examples/research_projects/gligen/demo.ipynb

@@ -26,7 +26,8 @@
     "%load_ext autoreload\n",
     "%autoreload 2\n",
     "\n",
-    "from diffusers import StableDiffusionGLIGENPipeline"
+    "import torch\n",
+    "from diffusers import StableDiffusionGLIGENTextImagePipeline, StableDiffusionGLIGENPipeline"
    ]
   },
   {
@@ -35,17 +36,16 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from transformers import CLIPTextModel, CLIPTokenizer\n",
-    "\n",
+    "import os\n",
     "import diffusers\n",
     "from diffusers import (\n",
     "    AutoencoderKL,\n",
     "    DDPMScheduler,\n",
-    "    EulerDiscreteScheduler,\n",
     "    UNet2DConditionModel,\n",
+    "    UniPCMultistepScheduler,\n",
+    "    EulerDiscreteScheduler,\n",
     ")\n",
-    "\n",
-    "\n",
+    "from transformers import CLIPImageProcessor, CLIPTextModel, CLIPTokenizer\n",
     "# pretrained_model_name_or_path = 'masterful/gligen-1-4-generation-text-box'\n",
     "\n",
     "pretrained_model_name_or_path = '/root/data/zhizhonghuang/checkpoints/models--masterful--gligen-1-4-generation-text-box/snapshots/d2820dc1e9ba6ca082051ce79cfd3eb468ae2c83'\n",
@@ -122,7 +122,6 @@
     "\n",
     "import numpy as np\n",
     "\n",
-    "\n",
     "boxes = np.array([x[1] for x in gen_boxes])\n",
     "boxes = boxes / 512\n",
     "boxes[:, 2] = boxes[:, 0] + boxes[:, 2]\n",