Usage Stats Collection (#2852)

2024-03-28 22:16:12 -07:00
parent 7bc94a0fdd
commit d8658c8cc1
15 changed files with 362 additions and 24 deletions
--- a/.buildkite/test-template.j2
+++ b/.buildkite/test-template.j2
@@ -53,6 +53,8 @@ steps:
                    nvidia.com/gpu: "{{ step.num_gpus or default_num_gpu }}"
                {% endif %}
                env:
+                  - name: VLLM_USAGE_SOURCE
+                    value: ci-test
                  - name: HF_TOKEN
                    valueFrom:
                      secretKeyRef: