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]