Update src/utils/metrics.py

Pavloveuge · yinggeh · web-flow · commit 41a39198a614 · 2025-07-24T20:46:49.000+03:00
Co-authored-by: Yingge He &lt;157551214+yinggeh@users.noreply.github.com&gt;
diff --git a/src/utils/metrics.py b/src/utils/metrics.py
@@ -366,7 +366,7 @@ def log(self, stats: VllmStats) -> None:
             (self.metrics.counter_preemption_tokens, stats.num_preemption_iter),
         ]
         histogram_metrics = [
-            (self.metrics.histogram_iteration_tokens_total, [stats.num_tokens_iter]),
+            (self.metrics.histogram_iteration_tokens, [stats.num_tokens_iter]),
             (
                 self.metrics.histogram_time_to_first_token,
                 stats.time_to_first_tokens_iter,

Original file line number	Diff line number	Diff line change
`@@ -366,7 +366,7 @@ def log(self, stats: VllmStats) -> None:`
`366`	`366`	`(self.metrics.counter_preemption_tokens, stats.num_preemption_iter),`
`367`	`367`	`]`
`368`	`368`	`histogram_metrics = [`
`369`		`- (self.metrics.histogram_iteration_tokens_total, [stats.num_tokens_iter]),`
	`369`	`+ (self.metrics.histogram_iteration_tokens, [stats.num_tokens_iter]),`
`370`	`370`	`(`
`371`	`371`	`self.metrics.histogram_time_to_first_token,`
`372`	`372`	`stats.time_to_first_tokens_iter,`