chore: add lm_head

zama-ai · Aug 6, 2024 · 7197c1d · 7197c1d
1 parent 956d0c4
commit 7197c1d
Show file tree

Hide file tree

Showing 2 changed files with 85 additions and 78 deletions.
diff --git a/script/make_utils/run_use_case_examples.sh b/script/make_utils/run_use_case_examples.sh
@@ -41,7 +41,7 @@ install_requirements() {
     local example_dir=$1
     if [ -f "${example_dir}/requirements.txt" ]; then
         pushd "$example_dir"
-        if pip install -r requirements.txt; then
+        if pip install -r requirements.txt --extra-index-url https://pypi.zama.ai/cpu; then
             echo "Requirements installed successfully."
         else
             echo "Failed to install requirements."