fix dependencies

codelion · codelion · commit ef0308dd4d82 · 2024-11-13T19:03:05.000+08:00
diff --git a/requirements.txt b/requirements.txt
@@ -18,4 +18,5 @@ presidio_anonymizer
 nbformat
 nbconvert
 ipython
-ipykernel
+ipykernel
+peft
diff --git 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': [
