Update Dockerfile

Dockerfile

@@ -61,28 +61,42 @@ RUN ln -sf /usr/bin/python3.11 /usr/bin/python && python -m pip install --upgrad
 # 1) Install JAX first
 RUN python -m pip install "jax[cuda12]==0.7.1" "jaxlib==0.7.1"
 
-# 2) Pin protobuf
+# 2) Install seqio dependencies manually (excluding tfds-nightly to avoid protobuf conflict)
+RUN python -m pip install \
+    absl-py clu sentencepiece tensorflow-text
+
+# 3) Install seqio without dependencies
+RUN python -m pip install --no-deps "seqio==0.0.11"
+
+# 4) Install correct protobuf version for tf-nightly
 RUN python -m pip install "protobuf>=6.31.1,<8.0.0"
-# 3) Install packages that depend on TF (let them bring their TF)
-RUN python -m pip install "seqio==0.0.11"
 
-# 4) NOW force install your desired TensorFlow version
-RUN python -m pip install --force-reinstall --no-deps \
+# 5) Install TensorFlow nightly without dependencies (to avoid protobuf downgrade)
+RUN python -m pip install --no-deps \
     "tf-nightly" \
     "tensorflow-hub==0.16.1"
 
-RUN python -c "import tensorflow as tf; print('='*50); print('FINAL 1TensorFlow version:', tf.__version__); print('='*50)"
-
-
-# 5) Install remaining packages
+# 6) Install remaining packages (consolidated, no duplicates)
 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
+    huggingface_hub gradio soxr cached-property
+
+# 7) Install airio
+RUN python -m pip install 'git+https://github.com/google/airio.git@main'
 
-RUN python -c "import tensorflow as tf; print('='*50); print('FINAL 2TensorFlow version:', tf.__version__); print('='*50)"
+# 8) Install t5x and flaxformer without dependencies
+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"
+
+# 9) Install magenta-rt without dependencies
+RUN python -m pip install --no-deps \
+    'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
 
+# 10) Verify TensorFlow installation
+RUN python -c "import tensorflow as tf; print('='*50); print('TensorFlow version:', tf.__version__); print('='*50)"
 
 RUN python -m pip install cached-property