test ci

Author: sbalandi

notebooks/ace-step-music-generation/ace-step-music-generation.ipynb

@@ -80,9 +80,9 @@
     "import platform\n",
     "from pathlib import Path\n",
     "\n",
-    "if not Path(\"ov_ace_helper.py\").exists():\n",
-    "    r = requests.get(url=\"https://raw.githubusercontent.com/openvinotoolkit/openvino_notebooks/latest/notebooks/ace-step-music-generation/ov_ace_helper.py\")\n",
-    "    open(\"ov_ace_helper.py\", \"w\").write(r.text)\n",
+    "if not Path(\"ov_ace_helper_test.py\").exists():\n",
+    "    r = requests.get(url=\"https://raw.githubusercontent.com/openvinotoolkit/openvino_notebooks/latest/notebooks/ace-step-music-generation/ov_ace_helper_test.py\")\n",
+    "    open(\"ov_ace_helper_test.py\", \"w\").write(r.text)\n",
     "\n",
     "if not Path(\"gradio_helper.py\").exists():\n",
     "    r = requests.get(url=\"https://raw.githubusercontent.com/openvinotoolkit/openvino_notebooks/latest/notebooks/ace-step-music-generation/gradio_helper.py\")\n",
@@ -390,7 +390,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": null,
    "id": "f9db0d8a",
    "metadata": {},
    "outputs": [
@@ -420,17 +420,74 @@
    ],
    "source": [
     "import nncf\n",
-    "from ov_ace_helper import convert_models\n",
     "\n",
     "if model_format.value == \"INT4\":\n",
     "    weights_compression_config = {\"mode\": nncf.CompressWeightsMode.INT4_ASYM, \"group_size\": 128, \"ratio\": 0.8}\n",
     "elif model_format.value == \"INT8\":\n",
     "    weights_compression_config = {\"mode\": nncf.CompressWeightsMode.INT8_ASYM}\n",
     "else:\n",
-    "    weights_compression_config = None\n",
+    "    weights_compression_config = None"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "c004eb47",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from ov_ace_helper_test import convert_text_encoder\n",
     "\n",
     "ov_converted_model_dir = \"ov_models\"\n",
-    "convert_models(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
+    "convert_text_encoder(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "0b44aa58",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from ov_ace_helper_test import convert_tokenizer_ov\n",
+    "\n",
+    "convert_tokenizer_ov(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d1f41973",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from ov_ace_helper_test import convert_vocoder\n",
+    "\n",
+    "convert_vocoder(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "d23e1ba9",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from ov_ace_helper_test import convert_transformer_models\n",
+    "\n",
+    "convert_transformer_models(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ce5ed806",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from ov_ace_helper_test import convert_dcae\n",
+    "\n",
+    "convert_dcae(pipeline, model_dir=ov_converted_model_dir, orig_checkpoint_path=checkpoint_dir, quantization_config=weights_compression_config)"
    ]
   },
   {
@@ -469,12 +526,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": null,
    "id": "a3d15fff",
    "metadata": {},
    "outputs": [],
    "source": [
-    "from ov_ace_helper import OVACEStepPipeline\n",
+    "from ov_ace_helper_test import OVACEStepPipeline\n",
     "\n",
     "ov_pipeline = OVACEStepPipeline()\n",
     "ov_pipeline.load_models(ov_models_path=ov_converted_model_dir, device=device.value)"
@@ -705,7 +762,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 18,
+   "execution_count": null,
    "id": "2f456bc4",
    "metadata": {},
    "outputs": [
@@ -721,7 +778,7 @@
     }
    ],
    "source": [
-    "from ov_ace_helper import convert_transformer_models\n",
+    "from ov_ace_helper_test import convert_transformer_models\n",
     "\n",
     "ov_models_lora = \"ov_models_lora\"\n",
     "\n",