fix dependenices on GPU for local inference

codelion · codelion · commit 917141f48163 · 2025-01-19T10:03:48.000+08:00
add protobuf
diff --git a/requirements.txt b/requirements.txt
@@ -21,4 +21,5 @@ ipython
 ipykernel
 peft
 bitsandbytes
-gradio
+gradio
+protobuf==3.20.3
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 setup(
     name="optillm",
-    version="0.0.24",
+    version="0.0.25",
     packages=find_packages(),
     py_modules=['optillm'],
     package_data={
@@ -34,6 +34,7 @@
         "peft",
         "bitsandbytes",
         "gradio",
+        "protobuf"
     ],
     entry_points={
         'console_scripts': [
