Skip to content

test(speechlm2): AUT-880 resolve eartts LM from mounted TestData to avoid HF 429#15917

Closed
svcnemo-autobot wants to merge 1 commit into
NVIDIA-NeMo:mainfrom
svcnemo-autobot:test/eartts-local-testdata-mount
Closed

test(speechlm2): AUT-880 resolve eartts LM from mounted TestData to avoid HF 429#15917
svcnemo-autobot wants to merge 1 commit into
NVIDIA-NeMo:mainfrom
svcnemo-autobot:test/eartts-local-testdata-mount

Conversation

@svcnemo-autobot

Copy link
Copy Markdown

Background

The L0_Unit_Tests_CPU_SpeechLM2 CI job intermittently fails at fixture setup of tests/collections/speechlm2/test_duplex_eartts.py::test_eartts_dataset. Instantiating DuplexEARTTS loads the nvidia/NVIDIA-Nemotron-Nano-9B-v2 tokenizer/backbone; in CI it reached the HuggingFace Hub unauthenticated and hit an HTTP 429 rate limit (We had to rate limit your IP).

What changed

  • Route the eartts test's pretrained_lm_name to a CI-cached copy under the mounted /home/TestData volume so the model resolves locally instead of via the Hub.

Details

  • tests/collections/speechlm2/test_duplex_eartts.py — the previous guard pointed at a bespoke path (/home/TestData/nvidia--NVIDIA-Nemotron-Nano-9B-v2/) that isn't staged on the runners. The rest of the speechlm2 suite (s2s, salm, voicechat, ...) resolves models under /home/TestData/speechlm/pretrained_models/..., which test-template mounts. The guard now prefers that canonical staged location and keeps the legacy top-level path as a fallback.
  • This is a stop-gap: it presumes the Nemotron model is staged at the canonical path on CI runners (as the other speechlm2 models are). If not yet staged, the runner-side TestData volume needs it added.

Tested

  • Local ast.parse / syntax check passes; longest added line 84 chars (repo limit 119). Full validation is the CI unit-test job with the mounted TestData.

The eartts unit tests instantiate DuplexEARTTS, which loads the
nvidia/NVIDIA-Nemotron-Nano-9B-v2 tokenizer/backbone. In CI the test fell
back to the HuggingFace Hub name and hit unauthenticated rate limits
(HTTP 429), erroring test_eartts_dataset at fixture setup.

Route the LM to a CI-cached copy under the mounted /home/TestData volume,
preferring the canonical staged location used by the rest of the speechlm2
suite (/home/TestData/speechlm/pretrained_models/...) and keeping the legacy
top-level path as a fallback, so the test resolves the model locally instead
of reaching the Hub.

Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
@svcnemo-autobot svcnemo-autobot added force-run-speech-ci python Pull requests that update python code labels Jul 16, 2026
@svcnemo-autobot

Copy link
Copy Markdown
Author

/ok to test 4464362

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

force-run-speech-ci python Pull requests that update python code

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants