diff --git a/setup.sh b/setup.sh index 4ebf960b73..b54f782ded 100755 --- a/setup.sh +++ b/setup.sh @@ -166,7 +166,6 @@ if [ "$IS_COLAB" = true ]; then -o "$LLAMA_CPP_DST" echo " Installing studio dependencies..." run_quiet "pip install studio" pip install -r "$SCRIPT_DIR/studio/backend/requirements/studio.txt" - run_quiet "pip install numpy==2.2.6" pip install --force-reinstall numpy==2.2.6 echo "✅ Python dependencies installed" else # Local: create venv (always start fresh to preserve correct install order) diff --git a/studio/backend/requirements/extras-no-deps.txt b/studio/backend/requirements/extras-no-deps.txt index c2a9c6bad8..b78b479e51 100644 --- a/studio/backend/requirements/extras-no-deps.txt +++ b/studio/backend/requirements/extras-no-deps.txt @@ -11,3 +11,4 @@ git+https://github.com/meta-pytorch/OpenEnv.git executorch==1.0.1 torch-c-dlpack-ext sentence_transformers==5.2.0 +transformers==4.57.1 diff --git a/studio/backend/requirements/overrides.txt b/studio/backend/requirements/overrides.txt index 02770f3953..6852f601ed 100644 --- a/studio/backend/requirements/overrides.txt +++ b/studio/backend/requirements/overrides.txt @@ -1,6 +1,5 @@ # Torch AO overrides (installed with --force-reinstall --no-cache-dir) torchao==0.14.0 -transformers==4.57.1 pytorch_tokenizers # Kernel packages