Skip to content

Commit

Permalink
fix: remove onnxruntime from required deps
Browse files Browse the repository at this point in the history
  • Loading branch information
CuriousDolphin committed Nov 29, 2024
1 parent 173ad86 commit cd219cb
Show file tree
Hide file tree
Showing 4 changed files with 16 additions and 6 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ jobs:
python-version: "3.12"
cache: "pip"
- name: Install dependencies
run: pip install .[dev,gpu]
run: pip install .[cpu,dev]
- name: Run test
run: make test
- name: Pytest coverage comment
Expand Down
15 changes: 12 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,12 +30,21 @@ To perform inference using TensorRT, ensure you have TensorRT version 10.5 insta

# Install

Nvidia GPU:
```bash
pip install .
pip install .[gpu]
```

Nvidia GPU,TensorRT:
```bash
pip install .[gpu,tensorrt]
```

CPU,COREML:
```bash
pip install .[cpu]
```

- local inference (CPU, COREML): `pip install .`
- local inference with GPU (CUDA): `pip install .[gpu]`


## 🤖 Cloud Inference
Expand Down
2 changes: 1 addition & 1 deletion notebooks/playground.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@
"metadata": {},
"outputs": [],
"source": [
"%pip install -e ..[dev,gpu,tensorrt]"
"%pip install -e ..[dev,gpu]"
]
},
{
Expand Down
3 changes: 2 additions & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@ dependencies = [
"tqdm~=4.67.1",
"numpy~=1.26.4",
"scipy~=1.14.1",
"onnxruntime==1.20.1"
]

authors = [{ name = "focoos.ai", email = "[email protected]" }]
Expand All @@ -43,4 +42,6 @@ dev = [
"gradio~=5.3.0",
"sniffio~=1.2.0",
]
cpu = ["onnxruntime==1.20.1"]
gpu = ["onnxruntime-gpu==1.20.1","nvidia-cuda-runtime-cu12==12.4.127"]
tensorrt = ["tensorrt==10.5.0"]

0 comments on commit cd219cb

Please sign in to comment.