From a7e69641171467f49487979b1ce29d03aaacf2ca Mon Sep 17 00:00:00 2001
From: Manan Shah <52329525+Manan17@users.noreply.github.com>
Date: Fri, 3 Apr 2026 00:03:35 -0500
Subject: [PATCH] Fix/gemma4 install script (#4815)

* transformer 5.5.0 has now been released

* fallback for python < 3.10
:
---
 install_gemma4_mlx.sh | 24 ++++++------------------
 1 file changed, 6 insertions(+), 18 deletions(-)

diff --git a/install_gemma4_mlx.sh b/install_gemma4_mlx.sh
index e06339e204..b653af9154 100755
--- a/install_gemma4_mlx.sh
+++ b/install_gemma4_mlx.sh
@@ -132,29 +132,17 @@ step "install" "installing mlx, mlx-lm..."
 uv pip install --python "$_VENV_PY" -q mlx mlx-lm 2>/dev/null
 substep "done"
 
-TRANSFORMERS_WHL="transformers-5.5.0-py3-none-any.whl"
-TRANSFORMERS_GH="git+https://github.com/huggingface/transformers.git@v5.5-release"
-
 step "install" "installing transformers>=5.5.0..."
-if uv pip install --python "$_VENV_PY" -q "$TRANSFORMERS_GH" 2>/dev/null; then
-    substep "installed from huggingface/transformers v5.5-release"
-elif uv pip install --python "$_VENV_PY" -q "transformers>=5.5.0" 2>/dev/null; then
+if uv pip install --python "$_VENV_PY" -q "transformers>=5.5.0" 2>/dev/null; then
     substep "installed from PyPI"
 else
-    substep "not on PyPI, trying unsloth branch..."
-    _whl_tmp=$(mktemp -d)/"${TRANSFORMERS_WHL}"
-    if curl -fsSL "${REPO_URL}/${TRANSFORMERS_WHL}" -o "$_whl_tmp" 2>/dev/null && \
-       uv pip install --python "$_VENV_PY" -q "$_whl_tmp" 2>/dev/null; then
-        substep "installed from branch ${BRANCH}"
-    elif [ -f "./${TRANSFORMERS_WHL}" ]; then
-        substep "using local ./${TRANSFORMERS_WHL}"
-        uv pip install --python "$_VENV_PY" -q "./${TRANSFORMERS_WHL}"
+    substep "PyPI install failed (Python <3.10?), trying GitHub..."
+    if uv pip install --python "$_VENV_PY" -q "git+https://github.com/huggingface/transformers.git@v5.5-release" 2>/dev/null; then
+        substep "installed from huggingface/transformers v5.5-release"
     else
-        rm -f "$_whl_tmp" 2>/dev/null
-        step "install" "skipping transformers — could not find >=5.5.0" "$C_WARN"
-        substep "tried: huggingface/transformers v5.5-release, PyPI, branch ${BRANCH}, local ./${TRANSFORMERS_WHL}"
+        step "warning" "could not install transformers>=5.5.0" "$C_WARN"
+        substep "tried: PyPI, huggingface/transformers v5.5-release"
     fi
-    rm -f "$_whl_tmp" 2>/dev/null
 fi
 
 # ── Find mlx-lm models directory ─────────────────────────────