update the dependencies.

microsoft · Jan 18, 2025 · d8396c7 · d8396c7
1 parent c117302
commit d8396c7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -20,7 +20,7 @@ classifiers = [
  # `pyproject.toml`'s `dependencies` field.
  # Make sure to keep this field in sync with what is in `requirements.txt`.
 dependencies = [
-    "torch",
+    "torch>=2.3.0",
     "datasets",
     "transformers>=4.45",
     "safetensors",
@@ -29,6 +29,8 @@ dependencies = [
     "gradio",
     "plotly==5.9.0",
     "pynvml",
+    "tqdm",
+    "sentence_transformers",
 ]
 
 [project.urls]

diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 cmake
 packaging
 setuptools>=64.0.0
-torch
+torch>=2.3.0
 wheel
 datasets
 transformers>=4.45