diff --git a/setup.py b/setup.py
index 35668bddb..7fb4c70be 100644
--- a/setup.py
+++ b/setup.py
@@ -75,6 +75,7 @@ base = [
     "braceexpand",
     "pyyaml",
     "pybind11",
+    "paddlelite",
     "paddleslim==2.3.4",
 ]