Skip to content

Commit

Permalink
Update the habana extra (#2077)
Browse files Browse the repository at this point in the history
  • Loading branch information
regisss committed Oct 22, 2024
1 parent 2ab96ec commit c5bc378
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -84,7 +84,7 @@
"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.43.0,<4.44.0"],
"habana": ["optimum-habana", "transformers>=4.45.0,<4.46.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"],
"graphcore": "optimum-graphcore",
Expand Down

0 comments on commit c5bc378

Please sign in to comment.