replace test model for minicpm

openvinotoolkit · Nov 29, 2024 · 4c2b526 · 4c2b526
1 parent e8cbd14
commit 4c2b526
Showing 1 changed file with 26 additions and 22 deletions.
diff --git a/notebooks/minicpm-v-multimodal-chatbot/minicpm-v-multimodal-chatbot.ipynb b/notebooks/minicpm-v-multimodal-chatbot/minicpm-v-multimodal-chatbot.ipynb
@@ -131,7 +131,11 @@
    "cell_type": "code",
    "execution_count": 3,
    "id": "82e846bb",
-   "metadata": {},
+   "metadata": {
+    "test_replace": {
+     "openbmb/MiniCPM-V-2_6": "katuni4ka/tiny-random-minicpmv-2_6"
+    }
+   },
    "outputs": [
     {
      "name": "stdout",
@@ -169,7 +173,7 @@
     "model_dir = Path(model_id.split(\"/\")[-1] + \"-ov\")\n",
     "\n",
     "if not model_dir.exists():\n",
-    "    optimum_cli(model_id, model_dir, additional_args={\"trust-remote-code\": \"\", \"weight-format\": \"fp16\"})\n",
+    "    optimum_cli(model_id, model_dir, additional_args={\"trust-remote-code\": \"\", \"weight-format\": \"fp16\", \"task\": \"image-text-to-text\"})\n",
     "    compress_lm_weights(model_dir)"
    ]
   },
@@ -204,22 +208,22 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": 1,
    "id": "626fef57",
    "metadata": {},
    "outputs": [
     {
      "data": {
       "application/vnd.jupyter.widget-view+json": {
-       "model_id": "84bba5eaf8cc4b7e97a5e5d3768146e3",
+       "model_id": "2362638a795340e6b3effb0805848768",
        "version_major": 2,
        "version_minor": 0
       },
       "text/plain": [
        "Dropdown(description='Device:', index=1, options=('CPU', 'AUTO'), value='AUTO')"
       ]
      },
-     "execution_count": 4,
+     "execution_count": 1,
      "metadata": {},
      "output_type": "execute_result"
     }
@@ -434,21 +438,7 @@
   "widgets": {
    "application/vnd.jupyter.widget-state+json": {
     "state": {
-     "41592555658f4eb69616c541894b88f0": {
-      "model_module": "@jupyter-widgets/controls",
-      "model_module_version": "2.0.0",
-      "model_name": "DescriptionStyleModel",
-      "state": {
-       "description_width": ""
-      }
-     },
-     "658a0c15a9cb47078c9c8647bff53d1e": {
-      "model_module": "@jupyter-widgets/base",
-      "model_module_version": "2.0.0",
-      "model_name": "LayoutModel",
-      "state": {}
-     },
-     "84bba5eaf8cc4b7e97a5e5d3768146e3": {
+     "2362638a795340e6b3effb0805848768": {
       "model_module": "@jupyter-widgets/controls",
       "model_module_version": "2.0.0",
       "model_name": "DropdownModel",
@@ -459,9 +449,23 @@
        ],
        "description": "Device:",
        "index": 1,
-       "layout": "IPY_MODEL_658a0c15a9cb47078c9c8647bff53d1e",
-       "style": "IPY_MODEL_41592555658f4eb69616c541894b88f0"
+       "layout": "IPY_MODEL_d737bcde20ac4ba38ecf0902eec67998",
+       "style": "IPY_MODEL_49b230bc877e422788033f49884843a2"
+      }
+     },
+     "49b230bc877e422788033f49884843a2": {
+      "model_module": "@jupyter-widgets/controls",
+      "model_module_version": "2.0.0",
+      "model_name": "DescriptionStyleModel",
+      "state": {
+       "description_width": ""
       }
+     },
+     "d737bcde20ac4ba38ecf0902eec67998": {
+      "model_module": "@jupyter-widgets/base",
+      "model_module_version": "2.0.0",
+      "model_name": "LayoutModel",
+      "state": {}
      }
     },
     "version_major": 2,