Update Dockerfile

Dockerfile

@@ -92,9 +92,10 @@ RUN python -m pip install --no-deps \
 RUN python -m pip install --no-deps \
     'git+https://github.com/magenta/magenta-realtime#egg=magenta_rt[gpu]'
 
-# 10) Upgrade to latest versions to support XlaCallModule v10
-RUN python -m pip install --upgrade --force-reinstall \
-    tf-nightly
+    # 10) Remove conflicting TensorFlow, then upgrade to latest
+RUN python -m pip uninstall -y tensorflow tensorflow-io-gcs-filesystem && \
+    python -m pip install --upgrade --force-reinstall tf-nightly
+
 
 # Also upgrade JAX to latest to ensure XLA compatibility
 RUN python -m pip install --upgrade "jax[cuda12]" jaxlib