Add Llama2HF dependencies

This commit is contained in:
Ales Kubicek 2023-09-25 23:47:45 +02:00 committed by Nils Blach
parent ef3ce13d39
commit 48f024297c

View File

@ -28,6 +28,9 @@ dependencies = [
"sympy>=1.12", "sympy>=1.12",
"torch>=2.0.1", "torch>=2.0.1",
"transformers>=4.31.0", "transformers>=4.31.0",
"accelerate>=0.21.0",
"bitsandbytes>=0.41.0",
"scipy",
] ]
[project.urls] [project.urls]