Merge branch 'main' into dependabot/pip/pre-commit-gte-3.8-and-lt-5

protectai · Oct 17, 2024 · 7098ad9 · 7098ad9
2 parents d497b36 + ae03db1
commit 7098ad9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml
@@ -25,7 +25,8 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install ".[dev, onnxruntime]"
+          python -m pip install ".[dev]"
+          python -m pip install ".[onnxruntime]"
 
       - name: Run pre-commit hooks
         uses: pre-commit/action@v3.0.1
diff --git a/pyproject.toml b/pyproject.toml
@@ -30,17 +30,17 @@ dependencies = [
   "regex==2024.9.11",
   "tiktoken>=0.5,<0.8",
   "torch>=2.4.0",
-  "transformers>=4.44.2",
+  "transformers==4.44.2",
   "structlog>=24",
   "oldest-supported-numpy"
 ]
 
 [project.optional-dependencies]
 onnxruntime = [
-  "optimum[onnxruntime]>=1.21",
+  "optimum[onnxruntime]==1.23.1",
 ]
 onnxruntime-gpu = [
-  "optimum[onnxruntime-gpu]>=1.21",
+  "optimum[onnxruntime-gpu]==1.23.1",
 ]
 docs-dev = [
   "mkdocs>=1.6,<2",