update

hf_demo.py

@@ -104,7 +104,14 @@ def demo_inference_stylization_artistic(ref_image, adapter_choice:str, prompt:st
                             from_scratch=False, device=device, weight_dtype=dtype)[0][adapter_scale][0]
     return pred_images
 
-
+@spaces.GPU
+def demo_inference_all(prompt:str, ref_image, adapter_choice="Andre Derain (fauvism)", seed:int=0, steps=20, guidance_scale=7.5, adapter_scale=1.0,start_noise=800):
+    results = []
+    results.append(demo_inference_gen_ori(prompt, seed, steps, guidance_scale))
+    results.append(demo_inference_gen_artistic(adapter_choice, prompt, seed, steps, guidance_scale, adapter_scale))
+    results.append(demo_inference_stylization_ori(ref_image, prompt, seed, steps, guidance_scale, start_noise))
+    results.append(demo_inference_stylization_artistic(ref_image, adapter_choice, prompt, seed, steps, guidance_scale, adapter_scale, start_noise))
+    return results
 
 
 block = gr.Blocks()
@@ -234,25 +241,33 @@ with block:
     examples = gr.Examples(
         examples=[
             ["Snow-covered trees with sunlight shining through",
-             "data/Snow-covered_trees_with_sunlight_shining_through.jpg"
+             "data/Snow-covered_trees_with_sunlight_shining_through.jpg",
              ],
             ["A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity.",
-             "data/0011772.jpg"],
+             "data/0011772.jpg",
+             ],
             ["a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys.",
-             "data/a_black_SUV_driving_down_a_highway_with_a_scenic_view_of_mountains_and_water_in_the_background._The_.jpg"
+             "data/a_black_SUV_driving_down_a_highway_with_a_scenic_view_of_mountains_and_water_in_the_background._The_.jpg",
              ],
             [
                 "A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape.",
-                "data/003904765.jpg"
+                "data/003904765.jpg",
             ]
 
         ],
         inputs=[
             text,
-            gallery_stylization_ref
+            gallery_stylization_ref,
+            adapter_choice,
+            seed,
+            steps,
+            scale,
+            adapter_scale,
+            start_timestep,
         ],
-        # cache_examples=True,
+        fn=demo_inference_all,
+        outputs=[gallery_gen_ori, gallery_gen_art, gallery_stylization_ori, gallery_stylization_art],
+        cache_examples=True,
     )
-
 block.launch()
 # block.launch(sharing=True)

hf_demo_test.ipynb

@@ -72,7 +72,7 @@
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "08154385ff1741b88c3bcf5b4b1e15b3",
+       "model_id": "21523c1459824498a59b1d237927a48e",
        "version_major": 2,
        "version_minor": 0
       },
@@ -91,7 +91,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 21,
    "id": "83916bc68ff5d914",
    "metadata": {
     "ExecuteTime": {
@@ -192,12 +192,20 @@
     "                            start_noise=start_noise, show=False, style_prompt=style_prompt, no_load=True,\n",
     "                            from_scratch=False, device=device, weight_dtype=dtype)[0][1.0][0]\n",
     "    return pred_images\n",
-    "\n"
+    "\n",
+    "\n",
+    "def demo_inference_all(prompt:str, ref_image, adapter_choice=\"Andre Derain (fauvism)\", seed:int=0, steps=20, guidance_scale=7.5, adapter_scale=1.0,start_noise=800):\n",
+    "    results = []\n",
+    "    results.append(demo_inference_gen_ori(prompt, seed, steps, guidance_scale))\n",
+    "    results.append(demo_inference_gen_artistic(adapter_choice, prompt, seed, steps, guidance_scale, adapter_scale))\n",
+    "    results.append(demo_inference_stylization_ori(ref_image, prompt, seed, steps, guidance_scale, start_noise))\n",
+    "    results.append(demo_inference_stylization_artistic(ref_image, adapter_choice, prompt, seed, steps, guidance_scale, adapter_scale, start_noise))\n",
+    "    return results"
    ]
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 22,
    "id": "aa33e9d104023847",
    "metadata": {
     "ExecuteTime": {
@@ -206,46 +214,192 @@
     }
    },
    "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/data/vision/torralba/selfmanaged/torralba/scratch/jomat/sam_dataset/miniforge3/envs/diffusion/lib/python3.9/site-packages/gradio/helpers.py:370: UserWarning: Examples will be cached but not all input components have example values. This may result in an exception being thrown by your function. If you do get an error while caching examples, make sure all of your inputs have example values for all of your examples or you provide default values for those particular parameters in your function.\n",
+      "  warnings.warn(\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Caching examples at: '/data/vision/torralba/selfmanaged/torralba/projects/jomat/hui/stable_diffusion/release/hf_demo/gradio_cached_examples/589'\n",
+      "Caching example 1/4\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:01\n",
+      "save dir: None\n",
+      "['Snow-covered trees with sunlight shining through'], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  8.41it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 2.58335280418396\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Train method: all_up\n",
+      "Rank: 1, Alpha: 1.0\n",
+      "create LoRA for U-Net: 123 modules.\n",
+      "Missing: <All keys matched successfully>\n",
+      "current time: 2024-12-14 01:19:04\n",
+      "save dir: None\n",
+      "['Snow-covered trees with sunlight shining through in the style of sks art'], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  8.20it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 2.6419241428375244\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n"
+     ]
+    },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Running on local URL:  http://127.0.0.1:7871\n",
-      "Running on public URL: https://e603b87db4de714e28.gradio.live\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:07\n",
+      "save dir: None\n",
+      "['Snow-covered trees with sunlight shining through'], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:01<00:00, 10.71it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 2.0403149127960205\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Train method: all_up\n",
+      "Rank: 1, Alpha: 1.0\n",
+      "create LoRA for U-Net: 123 modules.\n",
+      "Missing: <All keys matched successfully>\n",
+      "current time: 2024-12-14 01:19:09\n",
+      "save dir: None\n",
+      "['Snow-covered trees with sunlight shining through in the style of sks art'], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
       "\n",
-      "This share link expires in 72 hours. For free permanent hosting and GPU upgrades, run `gradio deploy` from Terminal to deploy to Spaces (https://huggingface.co/spaces)\n"
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  9.96it/s]"
      ]
     },
     {
-     "data": {
-      "text/html": [
-       "<div><iframe src=\"https://e603b87db4de714e28.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
-      ],
-      "text/plain": [
-       "<IPython.core.display.HTML object>"
-      ]
-     },
-     "metadata": {},
-     "output_type": "display_data"
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 2.1876306533813477\n",
+      "Caching example 2/4\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:12\n",
+      "save dir: None\n",
+      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity.\"], seed=0\n"
+     ]
     },
     {
-     "data": {
-      "text/plain": []
-     },
-     "execution_count": 17,
-     "metadata": {},
-     "output_type": "execute_result"
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  7.57it/s]"
+     ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 2.8535258769989014\n",
+      "Train method: all_up\n",
+      "Rank: 1, Alpha: 1.0\n",
+      "create LoRA for U-Net: 123 modules.\n",
+      "Missing: <All keys matched successfully>\n",
+      "current time: 2024-12-14 01:19:15\n",
+      "save dir: None\n",
+      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity in the style of sks art\"], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  7.09it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 3.0402302742004395\n",
       "Train method: None\n",
       "Rank: 1, Alpha: 1\n",
       "create LoRA for U-Net: 0 modules.\n",
-      "current time: 2024-12-14 00:07:17\n",
+      "current time: 2024-12-14 01:19:18\n",
       "save dir: None\n",
-      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity.\"], seed=881336985\n"
+      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity.\"], seed=0\n"
      ]
     },
     {
@@ -253,21 +407,43 @@
      "output_type": "stream",
      "text": [
       "\n",
-      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:01<00:00, 15.70it/s]"
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  9.29it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Time taken for one batch, Art Adapter scale=0.0: 1.4182789325714111\n",
+      "Time taken for one batch, Art Adapter scale=0.0: 2.33781099319458\n",
       "Train method: all_up\n",
       "Rank: 1, Alpha: 1.0\n",
       "create LoRA for U-Net: 123 modules.\n",
       "Missing: <All keys matched successfully>\n",
-      "current time: 2024-12-14 00:07:24\n",
+      "current time: 2024-12-14 01:19:21\n",
+      "save dir: None\n",
+      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity in the style of sks art\"], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  8.70it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 2.490582227706909\n",
+      "Caching example 3/4\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:24\n",
       "save dir: None\n",
-      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity in the style of sks art\"], seed=881336985\n"
+      "['a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys.'], seed=0\n"
      ]
     },
     {
@@ -275,77 +451,191 @@
      "output_type": "stream",
      "text": [
       "\n",
-      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:01<00:00, 15.38it/s]"
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:03<00:00,  6.56it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Time taken for one batch, Art Adapter scale=1: 1.4458158016204834\n",
+      "Time taken for one batch, Art Adapter scale=0.0: 3.2783615589141846\n",
       "Train method: all_up\n",
       "Rank: 1, Alpha: 1.0\n",
       "create LoRA for U-Net: 123 modules.\n",
       "Missing: <All keys matched successfully>\n",
-      "current time: 2024-12-14 00:07:32\n",
+      "current time: 2024-12-14 01:19:27\n",
       "save dir: None\n",
-      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity in the style of sks art\"], seed=881336985\n"
+      "['a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys in the style of sks art'], seed=0\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:01<00:00, 13.80it/s]"
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:03<00:00,  6.22it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Time taken for one batch, Art Adapter scale=1: 1.602182149887085\n"
+      "Time taken for one batch, Art Adapter scale=1.0: 3.4552266597747803\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:31\n",
+      "save dir: None\n",
+      "['a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys.'], seed=0\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\n"
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  8.18it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 2.6474156379699707\n",
       "Train method: all_up\n",
       "Rank: 1, Alpha: 1.0\n",
       "create LoRA for U-Net: 123 modules.\n",
       "Missing: <All keys matched successfully>\n",
-      "current time: 2024-12-14 00:07:42\n",
+      "current time: 2024-12-14 01:19:34\n",
       "save dir: None\n",
-      "[\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity in the style of sks art\"], seed=881336985\n"
+      "['a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys in the style of sks art'], seed=0\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "100%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:01<00:00, 12.56it/s]"
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  7.71it/s]"
      ]
     },
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Time taken for one batch, Art Adapter scale=1: 1.750511884689331\n"
+      "Time taken for one batch, Art Adapter scale=1.0: 2.803696870803833\n",
+      "Caching example 4/4\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:37\n",
+      "save dir: None\n",
+      "[\"A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape.\"], seed=0\n"
      ]
     },
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\n"
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:03<00:00,  5.88it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 3.648933172225952\n",
+      "Train method: all_up\n",
+      "Rank: 1, Alpha: 1.0\n",
+      "create LoRA for U-Net: 123 modules.\n",
+      "Missing: <All keys matched successfully>\n",
+      "current time: 2024-12-14 01:19:41\n",
+      "save dir: None\n",
+      "[\"A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape in the style of sks art\"], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:03<00:00,  5.57it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 3.8492958545684814\n",
+      "Train method: None\n",
+      "Rank: 1, Alpha: 1\n",
+      "create LoRA for U-Net: 0 modules.\n",
+      "current time: 2024-12-14 01:19:45\n",
+      "save dir: None\n",
+      "[\"A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape.\"], seed=0\n"
+     ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:02<00:00,  7.32it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=0.0: 2.9497482776641846\n",
+      "Train method: all_up\n",
+      "Rank: 1, Alpha: 1.0\n",
+      "create LoRA for U-Net: 123 modules.\n",
+      "Missing: <All keys matched successfully>\n",
+      "current time: 2024-12-14 01:19:49\n",
+      "save dir: None\n",
+      "[\"A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape in the style of sks art\"], seed=0\n"
      ]
+    },
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "\n",
+      "00%|██████████████████████████████████████████████████████████████████████████████████████████████████| 21/21 [00:03<00:00,  6.94it/s]"
+     ]
+    },
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Time taken for one batch, Art Adapter scale=1.0: 3.107145309448242\n",
+      "Running on local URL:  http://127.0.0.1:7865\n",
+      "Running on public URL: https://26432817d4a57fa05b.gradio.live\n",
+      "\n",
+      "This share link expires in 72 hours. For free permanent hosting and GPU upgrades, run `gradio deploy` from Terminal to deploy to Spaces (https://huggingface.co/spaces)\n"
+     ]
+    },
+    {
+     "data": {
+      "text/html": [
+       "<div><iframe src=\"https://26432817d4a57fa05b.gradio.live\" width=\"100%\" height=\"500\" allow=\"autoplay; camera; microphone; clipboard-read; clipboard-write;\" frameborder=\"0\" allowfullscreen></iframe></div>"
+      ],
+      "text/plain": [
+       "<IPython.core.display.HTML object>"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "text/plain": []
+     },
+     "execution_count": 22,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -435,7 +725,7 @@
     "                                height=\"auto\",\n",
     "                                scale=1,\n",
     "                            )\n",
-    "                    start_timestep = gr.Slider(label=\"Adapter Timestep\", minimum=0, maximum=1000, value=800, step=1)\n",
+    "                    start_timestep = gr.Slider(label=\"Timestep start from:\", minimum=0, maximum=1000, value=800, step=1)\n",
     "            with gr.Row():\n",
     "                btn_style_ori = gr.Button(\"Art-Free Stylize\", scale=1)\n",
     "                btn_style_art = gr.Button(\"Artistic Stylize\", scale=1)\n",
@@ -477,24 +767,33 @@
     "    examples = gr.Examples(\n",
     "        examples=[\n",
     "            [\"Snow-covered trees with sunlight shining through\",\n",
-    "             \"data/Snow-covered_trees_with_sunlight_shining_through.jpg\"\n",
+    "             \"data/Snow-covered_trees_with_sunlight_shining_through.jpg\",\n",
     "             ],\n",
     "            [\"A picturesque landscape showcasing a winding river cutting through a lush green valley, surrounded by rugged mountains under a clear blue sky. The mix of red and brown tones in the rocky hills adds to the region's natural beauty and diversity.\",\n",
-    "             \"data/0011772.jpg\"],\n",
+    "             \"data/0011772.jpg\",\n",
+    "            ],\n",
     "            [\"a black SUV driving down a highway with a scenic view of mountains and water in the background. The SUV is the main focus of the image, and it appears to be traveling at a moderate speed. The road is well-maintained and provides a smooth driving experience. The mountains and water create a picturesque backdrop, adding to the overall beauty of the scene. The image captures the essence of a leisurely road trip, with the SUV as the primary subject, highlighting the sense of adventure and exploration that comes with such journeys.\",\n",
-    "             \"data/a_black_SUV_driving_down_a_highway_with_a_scenic_view_of_mountains_and_water_in_the_background._The_.jpg\"\n",
+    "             \"data/a_black_SUV_driving_down_a_highway_with_a_scenic_view_of_mountains_and_water_in_the_background._The_.jpg\",\n",
     "             ],\n",
     "            [\n",
     "                \"A blue bench situated in a park, surrounded by trees and leaves. The bench is positioned under a tree, providing shade and a peaceful atmosphere. There are several benches in the park, with one being closer to the foreground and the others further in the background. A person can be seen in the distance, possibly enjoying the park or taking a walk. The overall scene is serene and inviting, with the bench serving as a focal point in the park's landscape.\",\n",
-    "                \"data/003904765.jpg\"\n",
+    "                \"data/003904765.jpg\",\n",
     "            ]\n",
     "\n",
     "        ],\n",
     "        inputs=[\n",
     "            text,\n",
-    "            gallery_stylization_ref\n",
+    "            gallery_stylization_ref,\n",
+    "            adapter_choice,\n",
+    "            seed,\n",
+    "            steps,\n",
+    "            scale,\n",
+    "            adapter_scale,\n",
+    "            start_timestep,            \n",
     "        ],\n",
-    "        # cache_examples=True,\n",
+    "        fn=demo_inference_all,\n",
+    "        outputs=[gallery_gen_ori, gallery_gen_art, gallery_stylization_ori, gallery_stylization_art],\n",
+    "        cache_examples=True,\n",
     "    )\n",
     "block.launch(share=True)"
    ]