Skip to content

Commit

Permalink
Update optimum intel extra (#1935)
Browse files Browse the repository at this point in the history
* update optimum intel extra

* add ipex extra
  • Loading branch information
echarlaix authored Jul 2, 2024
1 parent ae591be commit 4e01a4a
Showing 1 changed file with 5 additions and 4 deletions.
9 changes: 5 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"],
Expand Down

0 comments on commit 4e01a4a

Please sign in to comment.