From 48f024297c071e9302b8d07bb1e1c37ad014dc4c Mon Sep 17 00:00:00 2001 From: Ales Kubicek Date: Mon, 25 Sep 2023 23:47:45 +0200 Subject: [PATCH] Add Llama2HF dependencies --- pyproject.toml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/pyproject.toml b/pyproject.toml index 58da8cd..202f8c1 100644 --- 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]