Skip to content

Commit 00aaec2

Browse files
tjohnson31415njhill
authored andcommitted
fix: add extra for CPU onnx
Apparently, optional = true only works if there is an extras package that installs the dependency. Adding this additional extra makes it so that onnxruntime and onnxruntime-gpu are not both installed together. Signed-off-by: Travis Johnson <[email protected]>
1 parent 5e93dbc commit 00aaec2

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

server/poetry.lock

Lines changed: 2 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

server/pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ cryptography = ">=41.0.3"
3434
[tool.poetry.extras]
3535
accelerate = ["accelerate"]
3636
bnb = ["bitsandbytes"]
37+
onnx = ["optimum", "onnxruntime", "onnx"]
3738
onnx-gpu = ["optimum", "onnxruntime-gpu", "onnx"]
3839

3940
[tool.poetry.group.dev.dependencies]

0 commit comments

Comments
 (0)