Fixed pre-ci test error

Signed-off-by: Cheng, Penghui <[email protected]>
PenghuiCheng · Mar 13, 2024 · 3ca3f60 · 3ca3f60
1 parent 82c27dd
commit 3ca3f60
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/setup.py b/setup.py
@@ -34,6 +34,7 @@
     "rjieba",
     "timm",
     "invisible-watermark>=0.2.0",
+    "cmake>=3.16",
     "intel-extension-for-transformers>=1.3",
     "peft",
     "auto-gptq",