From 3dfd73da8e50789b5544fed547004df27073c601 Mon Sep 17 00:00:00 2001 From: Nicolas Oliver Date: Thu, 16 May 2024 09:44:15 -0700 Subject: [PATCH 1/2] Update quantized_generation_demo.ipynb Fix version of torch to 2.0.1. See https://github.com/pytorch/pytorch/issues/125109 --- notebooks/openvino/quantized_generation_demo.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/openvino/quantized_generation_demo.ipynb b/notebooks/openvino/quantized_generation_demo.ipynb index 7671064088..9ca488b7d1 100644 --- 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" ] }, { From 64753d6e29efee7c649b35700d75aafb4fbde3bb Mon Sep 17 00:00:00 2001 From: Nicolas Oliver Date: Fri, 17 May 2024 09:28:25 -0700 Subject: [PATCH 2/2] Update quantized_generation_demo.ipynb Bump version of torch to 2.2.2 --- notebooks/openvino/quantized_generation_demo.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/openvino/quantized_generation_demo.ipynb b/notebooks/openvino/quantized_generation_demo.ipynb index 9ca488b7d1..5673243cb2 100644 --- 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==2.0.1" + "# ! pip install optimum[openvino,nncf] torch==2.2.2" ] }, {