Update quantized_generation_demo.ipynb

Fix version of torch to 2.0.1. See pytorch/pytorch#125109
dnoliver · May 16, 2024 · 3dfd73d · 3dfd73d
1 parent c30d488
commit 3dfd73d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/openvino/quantized_generation_demo.ipynb b/notebooks/openvino/quantized_generation_demo.ipynb
@@ -32,7 +32,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# ! pip install optimum[openvino,nncf] torch"
+    "# ! pip install optimum[openvino,nncf] torch==2.0.1"
    ]
   },
   {