Skip to content

Commit

Permalink
pin numpy 1 in exporters as well
Browse files Browse the repository at this point in the history
  • Loading branch information
IlyasMoutawwakil committed Jun 25, 2024
1 parent f2dec8c commit 9fcdf62
Showing 1 changed file with 9 additions and 4 deletions.
13 changes: 9 additions & 4 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,18 +53,23 @@
"datasets>=1.2.1",
"evaluate",
"protobuf>=3.20.1",
"numpy<=1.26.4",
"numpy<=1.26.4", # onnxruntime is still not compatible with numpy 2
],
"onnxruntime-gpu": [
"onnx",
"onnxruntime-gpu>=1.11.0",
"datasets>=1.2.1",
"evaluate",
"protobuf>=3.20.1",
"numpy<=1.26.4", # onnxruntime is still not compatible with numpy 2
"accelerate", # ORTTrainer requires it.
"numpy<=1.26.4",
],
"exporters": ["onnx", "onnxruntime", "timm"],
"exporters": [
"onnx",
"onnxruntime",
"timm",
"numpy<=1.26.4", # onnxruntime is still not compatible with numpy 2
],
"exporters-gpu": ["onnx", "onnxruntime-gpu", "timm"],
"exporters-tf": [
"tensorflow>=2.4,<=2.12.1",
Expand All @@ -81,10 +86,10 @@
"openvino": "optimum-intel[openvino]>=1.16.0",
"nncf": "optimum-intel[nncf]>=1.16.0",
"neural-compressor": "optimum-intel[neural-compressor]>=1.16.0",
"graphcore": "optimum-graphcore",
"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"],
"graphcore": "optimum-graphcore",
"furiosa": "optimum-furiosa",
"amd": "optimum-amd",
"dev": TESTS_REQUIRE + QUALITY_REQUIRE,
Expand Down

0 comments on commit 9fcdf62

Please sign in to comment.