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': [