Add metal tests

xorbitsai · Jul 4, 2024 · a725932 · a725932
1 parent 69a8d42
commit a725932
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/.github/workflows/python.yaml b/.github/workflows/python.yaml
@@ -110,6 +110,9 @@ jobs:
             sudo rm -rf "/usr/local/share/boost"
             sudo rm -rf "$AGENT_TOOLSDIRECTORY"
           fi
+          if [ "$MODULE" == "metal" ]; then
+            pip install mlx-lm
+          fi
           pip install "llama-cpp-python==0.2.77"
           pip install transformers
           pip install attrdict