diff --git a/deps/JetStream b/deps/JetStream
index d462ca9..8eb7c5c 160000
--- a/deps/JetStream
+++ b/deps/JetStream
@@ -1 +1 @@
-Subproject commit d462ca9bbc55531bbe785203cb076e7797250f2a
+Subproject commit 8eb7c5c269725afd98485535ebea134cc625bb8a
diff --git a/deps/xla b/deps/xla
index fb2d4e1..cff9f4e 160000
--- a/deps/xla
+++ b/deps/xla
@@ -1 +1 @@
-Subproject commit fb2d4e1464dfd96f38a343c0e6f512629e28b48c
+Subproject commit cff9f4e0660df37f897f65aaf8d754981c2ba709
diff --git a/jetstream_pt/cli.py b/jetstream_pt/cli.py
index 66c378a..3b61780 100644
--- a/jetstream_pt/cli.py
+++ b/jetstream_pt/cli.py
@@ -110,7 +110,9 @@ def serve():
   metrics_server_config: MetricsServerConfig | None = None
   if FLAGS.prometheus_port != 0:
     if 1 <= FLAGS.prometheus_port <= 65535:
-      metrics_server_config = MetricsServerConfig(port=FLAGS.prometheus_port)
+      metrics_server_config = MetricsServerConfig(
+          port=FLAGS.prometheus_port, model_name=FLAGS.model_name
+      )
     else:
       raise ValueError(
           f"Invalid port number: {FLAGS.prometheus_port}. Port must be between 1 and 65535."