Skip to content
This repository was archived by the owner on Oct 11, 2024. It is now read-only.

Force-disable upstream tracking #321

Merged
merged 6 commits into from
Jun 22, 2024
Merged

Force-disable upstream tracking #321

merged 6 commits into from
Jun 22, 2024

Conversation

dbarbuzzi
Copy link

This PR force-disables upstream tracking by forcing VLLM_DO_NOT_TRACK and VLLM_NO_USAGE_STATS to be True (disable data collection) rather than inspecting the environment variable.

An alternative approach could be to update the definition of is_usage_stats_enabled (in vllm/usage/usage_lib.py:29) to always return False and set the global _USAGE_STATS_ENABLED to be False.

@dbarbuzzi dbarbuzzi self-assigned this Jun 21, 2024
@dbarbuzzi
Copy link
Author

@dhuangnm
Copy link
Member

Should the default values be True too?

VLLM_NO_USAGE_STATS: bool = False
VLLM_DO_NOT_TRACK: bool = False

@dbarbuzzi
Copy link
Author

Should the default values be True too?

That section is purely for type-checking (as in, it is never actually executed because it’s protected by the if TYPE_CHECKING clause), however, I think this is a good change for clarity, so I pushed it.

@robertgshaw2-redhat robertgshaw2-redhat merged commit d14c46e into main Jun 22, 2024
37 checks passed
@robertgshaw2-redhat robertgshaw2-redhat deleted the disable-tracking branch June 22, 2024 11:32
derekk-nm pushed a commit that referenced this pull request Jun 24, 2024
This PR force-disables upstream tracking by forcing `VLLM_DO_NOT_TRACK`
and `VLLM_NO_USAGE_STATS` to be `True` (disable data collection) rather
than inspecting the environment variable.

An alternative approach could be to update the definition of
`is_usage_stats_enabled` (in
[`vllm/usage/usage_lib.py:29`](https://github.com/neuralmagic/nm-vllm/blob/main/vllm/usage/usage_lib.py#L29))
to always return `False` and set the `global _USAGE_STATS_ENABLED` to be
`False`.
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants