📌 Pin breaking import changes for torch 2.3.0

Signed-off-by: Evaline Ju <[email protected]>
flaviabeo · Apr 24, 2024 · cb9a51a · cb9a51a
1 parent 4cc6b88
commit cb9a51a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ dependencies = [
  "scipy>=1.8.1",
  "sentence-transformers>=2.3.1,<2.4.0",
  "tokenizers>=0.13.3",
- "torch>=2.0.1",
+ "torch>=2.0.1,<2.3.0",
  "tqdm>=4.65.0",
  "transformers>=4.32.0",
  "peft==0.6.0",