Update Dockerfile

Dockerfile

@@ -58,40 +58,38 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN ln -sf /usr/bin/python3.11 /usr/bin/python && python -m pip install --upgrade pip
 
 # --- Python deps (pin order matters!) ---
-# 1) Install JAX
+# 1) Install JAX first
 RUN python -m pip install "jax[cuda12]==0.7.1" "jaxlib==0.7.1"
 
-# 2) Pin protobuf BEFORE anything else
+# 2) Pin protobuf
 RUN python -m pip install "protobuf>=4.21.0,<5.0.0"
 
-# 3) Install seqio
+# 3) Install packages that depend on TF (let them bring their TF)
 RUN python -m pip install "seqio==0.0.11"
 
-RUN sed -i '/import tensorflow_text as tf_text/d' /usr/local/lib/python3.11/dist-packages/seqio/vocabularies.py
-
-# 4) Install TF stable (should respect protobuf constraint)
-RUN python -m pip install \
+# 4) NOW force install your desired TensorFlow version
+RUN python -m pip install --force-reinstall --no-deps \
     "tf-nightly" \
     "tensorflow-hub==0.16.1"
 
-# 5) Install magenta-rt without deps
-RUN python -m pip install --no-deps \
-    'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
-
-    
-
-# 6) Rest of deps
+# 5) Install remaining packages
 RUN python -m pip install \
     tf2jax gin-config librosa resampy soundfile \
     fastapi uvicorn[standard] python-multipart pyloudnorm \
     google-cloud-storage "numpy==2.1.3" \
     huggingface_hub gradio soxr
 
-# 7) t5x/flaxformer last
-RUN python -m pip install \
+# 6) t5x/flaxformer with --no-deps
+RUN python -m pip install --no-deps \
     "t5x @ git+https://github.com/google-research/t5x.git@92c5b46" \
     "flaxformer @ git+https://github.com/google/flaxformer@399ea3a"
 
+# 7) magenta-rt last with --no-deps
+RUN python -m pip install --no-deps \
+    'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
+
+# 8) Verify final TensorFlow version
+RUN python -c "import tensorflow as tf; print('='*50); print('FINAL TensorFlow version:', tf.__version__); print('='*50)"
 
 RUN python -m pip install huggingface_hub