Skip to content

Commit

Permalink
fix: bundle cuda-runtime and onnxruntime dependencies (#19)
Browse files Browse the repository at this point in the history
  • Loading branch information
CuriousDolphin authored Nov 29, 2024
1 parent df77b86 commit 173ad86
Show file tree
Hide file tree
Showing 4 changed files with 6 additions and 15 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/release.yml
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ jobs:
file_pattern: pyproject.toml
# Step 4: Attendere che il commit sia sincronizzato su GitHub
- name: Wait for commit sync
run: sleep 20
run: sleep 30
- uses: actions/checkout@v4
# Step 5: Creare e taggare la nuova versione
- name: Tag version
Expand Down
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]
run: pip install .[dev,gpu]
- name: Run test
run: make test
- name: Pytest coverage comment
Expand Down
13 changes: 2 additions & 11 deletions notebooks/playground.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -13,16 +13,7 @@
"metadata": {},
"outputs": [],
"source": [
"%pip uninstall -y onnxruntime-gpu"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"%pip install -e ..[dev,gpu]"
"%pip install -e ..[dev,gpu,tensorrt]"
]
},
{
Expand All @@ -42,7 +33,7 @@
"from pprint import pprint\n",
"import os\n",
"\n",
"focoos = Focoos(api_key=os.getenv(\"FOCOOS_API_KEY\"), host_url=LOCAL_API_URL)\n",
"focoos = Focoos(api_key=os.getenv(\"FOCOOS_API_KEY\"), host_url=DEV_API_URL)\n",
"\n",
"pprint(focoos.list_focoos_models())"
]
Expand Down
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ include = ["focoos**"]

[project]
name = "focoos"
version = "0.1.1"
version = "0.1.2"
description = "Focoos SDK"
readme = "README.md"
requires-python = ">=3.10"
Expand Down Expand Up @@ -43,4 +43,4 @@ dev = [
"gradio~=5.3.0",
"sniffio~=1.2.0",
]
gpu = ["onnxruntime-gpu==1.20.1"]
gpu = ["onnxruntime-gpu==1.20.1","nvidia-cuda-runtime-cu12==12.4.127"]

0 comments on commit 173ad86

Please sign in to comment.