Add Llama2HF dependencies
This commit is contained in:
parent
ef3ce13d39
commit
48f024297c
@ -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]
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user