From 4e01a4a948cf48a9152f86349e82ea6cc72a0d03 Mon Sep 17 00:00:00 2001 From: Ella Charlaix <80481427+echarlaix@users.noreply.github.com> Date: Tue, 2 Jul 2024 14:06:45 +0200 Subject: [PATCH] Update optimum intel extra (#1935) * update optimum intel extra * add ipex extra --- setup.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index cc88760d614..519d04eab8e 100644 --- a/setup.py +++ b/setup.py @@ -75,10 +75,11 @@ "transformers[sentencepiece]>=4.26.0,<4.38.0", ], "diffusers": ["diffusers"], - "intel": "optimum-intel>=1.16.0", - "openvino": "optimum-intel[openvino]>=1.16.0", - "nncf": "optimum-intel[nncf]>=1.16.0", - "neural-compressor": "optimum-intel[neural-compressor]>=1.16.0", + "intel": "optimum-intel>=1.18.0", + "openvino": "optimum-intel[openvino]>=1.18.0", + "nncf": "optimum-intel[nncf]>=1.18.0", + "neural-compressor": "optimum-intel[neural-compressor]>=1.18.0", + "ipex": "optimum-intel[ipex]>=1.18.0", "habana": ["optimum-habana", "transformers >= 4.38.0, < 4.39.0"], "neuron": ["optimum-neuron[neuron]>=0.0.20", "transformers >= 4.36.2, < 4.42.0"], "neuronx": ["optimum-neuron[neuronx]>=0.0.20", "transformers >= 4.36.2, < 4.42.0"],