You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The warning message in the code even references the wrong variable name, printing TOKENI...LISM=false to suppress this warning, which is misleading to users.
Steps to Reproduce
fromllmcompressor.entrypoints.oneshotimportTOKENIZERS_PARALLELISM_ENVprint(repr(TOKENIZERS_PARALLELISM_ENV)) # prints 'TOKENI...LISM' instead of 'TOKENIZERS_PARALLELISM'
The associated unit tests in tests/llmcompressor/transformers/oneshot/test_tokenizer_parallelism.py check that os.environ[_TOKENIZERS_PARALLELISM_ENV] is set, but since the constant name is wrong, those tests pass while the real env var TOKENIZERS_PARALLELISM remains unset.
Fix
Change line 35 of src/llmcompressor/entrypoints/oneshot.py from:
Bug Description
The constant
TOKENIZERS_PARALLELISM_ENVinsrc/llmcompressor/entrypoints/oneshot.pyhas been assigned a corrupted/truncated value:It should be:
Impact
This bug means that:
Oneshot.__init__method sets the environment variableTOKENI...LISM=falseinstead of the correctTOKENIZERS_PARALLELISM=false.TOKENI...LISM=false to suppress this warning, which is misleading to users.Steps to Reproduce
The associated unit tests in
tests/llmcompressor/transformers/oneshot/test_tokenizer_parallelism.pycheck thatos.environ[_TOKENIZERS_PARALLELISM_ENV]is set, but since the constant name is wrong, those tests pass while the real env varTOKENIZERS_PARALLELISMremains unset.Fix
Change line 35 of
src/llmcompressor/entrypoints/oneshot.pyfrom:to:
This was likely introduced by a find/replace or editor issue that truncated the string value.
Environment
src/llmcompressor/entrypoints/oneshot.py, line 35