Add Llama2HF dependencies

2023-09-26 15:15:07 +02:00
parent ef3ce13d39
commit 48f024297c
1 changed files with 3 additions and 0 deletions
@@ -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]