Fix venv setup for brevitas (#1779)

nod-ai · Aug 21, 2023 · b87efe7 · b87efe7
1 parent 82b462d
commit b87efe7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup_venv.sh b/setup_venv.sh
@@ -146,7 +146,7 @@ if [[ $(uname -s) = 'Linux' && ! -z "${BENCHMARK}" ]]; then
 fi
 
 if [[ -z "${NO_BREVITAS}" ]]; then
-  $PYTHON -m pip install git+https://github.com/Xilinx/brevitas.git@llm
+  $PYTHON -m pip install git+https://github.com/Xilinx/brevitas.git@dev
 fi
 
 if [[ -z "${CONDA_PREFIX}" && "$SKIP_VENV" != "1" ]]; then