Merge pull request #2741 from AI-Hypercomputer:qwen3_colab_fix

PiperOrigin-RevId: 836502158

Author: Google-ML-Automation

src/MaxText/examples/sft_qwen3_demo.ipynb

@@ -89,6 +89,13 @@
     "## Installation: MaxText & Other Dependencies"
    ]
   },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "**⚠️ Note:** The installation process in following cell may take a few minutes to complete. Please be patient."
+   ]
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -108,7 +115,8 @@
     "!python3 -m MaxText.install_maxtext_extra_deps\n",
     "\n",
     "# Install vLLM for Jax and TPUs\n",
-    "!uv pip install vllm-tpu"
+    "!uv pip install vllm-tpu\n",
+    "!uv pip install --no-deps qwix==0.1.4"
    ]
   },
   {
@@ -154,6 +162,8 @@
     "from MaxText.integration.tunix.tunix_adapter import TunixMaxTextAdapter\n",
     "from MaxText.sft import sft_trainer\n",
     "\n",
+    "# Suppress vLLM logging with a severity level below ERROR\n",
+    "os.environ[\"VLLM_LOGGING_LEVEL\"] = \"ERROR\"\n",
     "from tunix.rl.rollout.vllm_rollout import VllmRollout\n",
     "\n",
     "from datetime import datetime\n",