Add vibevoice/pytorch-1.5B single-device inference test config#5341
Draft
saiarthiraguram wants to merge 1 commit into
Draft
Add vibevoice/pytorch-1.5B single-device inference test config#5341saiarthiraguram wants to merge 1 commit into
saiarthiraguram wants to merge 1 commit into
Conversation
Mark VibeVoice-1.5B as EXPECTED_PASSING on n150 after single-device bringup. Restricted to n150 (the verified arch); the model runs via the registry-driven test_all_models_torch path. Default PCC comparison passes, so no assert_pcc override is needed. Loader added in tt-forge-models (separate PR). Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Ticket
Link to Github Issue
Problem description
microsoft/VibeVoice-1.5Bwas brought up single-device on n150 (inference), but it is not registered in the inference test config, so it is not exercised by the test suite. The loader for it is added in a companion tt-forge-models PR (branchsai_arthi_raguram/vibe_voice), which must land and be uplifted before this entry resolves.What's changed
Registers VibeVoice-1.5B for single-device inference testing:
supported_archs: ["n150"]— bringup was verified on n150 only; restrict to that arch (matches the single-arch convention used by e.g.falcon/pytorch-3_*).status: EXPECTED_PASSING— the registry-driventest_all_models_torchrun passes with the default PCC comparison.assert_pcc/required_pccoverride — default PCC assertion already passes, so there's nothing to mask.tests/runner/test_models.py::test_all_models_torch[...]path; there is no per-modeltest_*.py, and the loader'sModelInfocarries nobringup_statusfield. The YAMLstatusis the single source of truth.Impact: VibeVoice-1.5B is now covered by single-device inference CI on n150.
Provenance: tt-xla
cb89a724b; tt-forge-modelsf96d6a82a0at bringup.Checklist
test_all_models_torch[vibevoice/pytorch-1.5B-single_device-inference]→1 passed in 210.41son n150.Logs
bringup_steps.txt
iter_2_run.log
model_overview.md