From 016be16299247c050d2dc83f766ad06aa4c28e34 Mon Sep 17 00:00:00 2001 From: Asankhaya Sharma Date: Wed, 13 Nov 2024 19:09:21 +0800 Subject: [PATCH] add dependency and bump version --- requirements.txt | 3 ++- setup.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 2f896ef..1d710c6 100644 --- a/requirements.txt +++ b/requirements.txt @@ -19,4 +19,5 @@ nbformat nbconvert ipython ipykernel -peft \ No newline at end of file +peft +bitsandbytes \ No newline at end of file diff --git a/setup.py b/setup.py index b9081ae..d880f03 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="optillm", - version="0.0.11", + version="0.0.12", packages=find_packages(), py_modules=['optillm'], package_data={ @@ -32,6 +32,7 @@ "ipython", "ipykernel", "peft", + "bitsandbytes", ], entry_points={ 'console_scripts': [