Fix Transformers modeling backend usage stats (#47472)
Harry Mellor committed
Jul 2, 2026 at 20:51 UTC
e24d1b24fe96a56ba8b0d653efa076d03eb95d6c
1 file changed
+8
-1
vllm/v1/utils.py
+8
-1
@@ -689,8 +689,15 @@ def report_usage_stats(
689
else None
690
)
691
692
+ if model_config.using_transformers_backend():
693
+ backend_cls = model_config._model_info.architecture
694
+ # Show what was wrapped e.g. TransformersForCausalLM(Starcoder2ForCausalLM)
695
+ architecture = f"{backend_cls}({model_config.architectures[0]})"
696
+ else:
697
+ architecture = get_architecture_class_name(model_config)
698
+
699
usage_message.report_usage(
693
- get_architecture_class_name(model_config),
700
+ architecture,
701
usage_context,
702
extra_kvs={
703
# Common configuration