Update comments

Author: yinggeh

src/metrics.py

@@ -234,7 +234,7 @@ def __init__(self, labels):
 
 
 class VllmStatLogger(VllmStatLoggerBase):
-    """StatLoggeris used as adapter between vLLM stats collector and Triton metrics provider."""
+    """StatLogger is used as an adapter between vLLM stats collector and Triton metrics provider."""
 
     # local_interval not used here. It's for vLLM logs to stdout.
     def __init__(self, labels: Dict, local_interval: float = 0) -> None:

src/model.py

@@ -101,6 +101,7 @@ def auto_complete_config(auto_complete_model_config):
         return auto_complete_model_config
 
     def initialize(self, args):
+        print(args)
         self.args = args
         self.logger = pb_utils.Logger
         self.model_config = json.loads(args["model_config"])
@@ -154,8 +155,11 @@ def init_engine(self):
         )
 
         # Create vLLM custom Metrics
-        labels = {"model": "vllm_metrics", "version": "1"}
-        logger = VllmStatLogger(vllm_labels=labels)
+        labels = {
+            "model": self.args["model_name"],
+            "version": self.args["model_version"],
+        }
+        logger = VllmStatLogger(labels=labels)
         self.llm_engine.add_logger("triton", logger)
 
     def setup_lora(self):