Add 'model_name' label to metrics and add MetricsServerConfig (#209)

Bslabe123 · web-flow · commit f4d775358c3e · 2025-03-27T13:55:55.000-07:00
* first commit

* config -&gt; flags

* config -&gt; flags

* fmt

* first commit

* missing comma

* update submodules
diff --git a/deps/JetStream b/deps/JetStream
@@ -1 +1 @@
-Subproject commit d462ca9bbc55531bbe785203cb076e7797250f2a
+Subproject commit 8eb7c5c269725afd98485535ebea134cc625bb8a
diff --git 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
@@ -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."
