update saving directory

huggingface · Mar 18, 2024 · eaf327c · eaf327c
1 parent d1ee29f
commit eaf327c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebooks/openvino/quantized_generation_demo.ipynb b/notebooks/openvino/quantized_generation_demo.ipynb
@@ -76,7 +76,7 @@
    "outputs": [],
    "source": [
     "model_name = 'microsoft/phi-2'\n",
-    "save_name = './phi-2-woq4'\n",
+    "save_name = model_name.split(\"/\")[-1] + '_openvino'\n",
     "precision = 'f32'\n",
     "quantization_config = OVWeightQuantizationConfig(\n",
     "    bits=4,\n",
@@ -575,7 +575,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.13"
+   "version": "3.9.18"
   }
  },
  "nbformat": 4,