diff --git a/requirements.txt b/requirements.txt index 8d61860..2f896ef 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,4 +18,5 @@ presidio_anonymizer nbformat nbconvert ipython -ipykernel \ No newline at end of file +ipykernel +peft \ No newline at end of file diff --git a/setup.py b/setup.py index 9a950f6..b9081ae 100644 --- a/setup.py +++ b/setup.py @@ -2,7 +2,7 @@ setup( name="optillm", - version="0.0.10", + version="0.0.11", packages=find_packages(), py_modules=['optillm'], package_data={ @@ -31,6 +31,7 @@ "nbformat", "ipython", "ipykernel", + "peft", ], entry_points={ 'console_scripts': [