Add Llama2HF dependencies

2023-09-25 23:47:45 +02:00 · 2023-09-25 23:47:45 +02:00 · 48f024297c
commit 48f024297c
parent ef3ce13d39
1 changed files with 3 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@ -28,6 +28,9 @@ dependencies = [
  "sympy>=1.12",
  "torch>=2.0.1",
  "transformers>=4.31.0",
+  "accelerate>=0.21.0",
+  "bitsandbytes>=0.41.0",
+  "scipy",
 ]

 [project.urls]