Skip to content

Conversation

@blisc
Copy link
Collaborator

@blisc blisc commented Dec 19, 2025

Important

The Update branch button must only be pressed in very rare occassions.
An outdated branch is never blocking the merge of a PR.
Please reach out to the automation team before pressing that button.

What does this PR do ?

Removes use of torchaudio.transforms and moves transforms inside of NeMo.
NOTE: we will use torchsquirm in nemo/collections/audio/metrics/squim.py and nemo/collections/tts/models/magpietts_preference_optimization.py

Collection: audio, asr, tts

Changelog

  • Move frequently used torchaudio transform into NeMo

PR Type:

  • New Feature
  • Bugfix
  • Documentation

Signed-off-by: Jason <[email protected]>
Signed-off-by: Jason <[email protected]>
nithinraok
nithinraok previously approved these changes Jan 8, 2026
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@chtruong814 could you help review the docker file changes.

@nithinraok
Copy link
Member

@blisc I think updates to docs/source/speechlm2/intro.rst are missing

@blisc
Copy link
Collaborator Author

blisc commented Jan 8, 2026

@blisc I think updates to docs/source/speechlm2/intro.rst are missing

I don't know if there is an equivalent to torchaudio.load() so I'm deferring making that change. @PiotrDabkowski Do you have a recommendation on how to update that docs?

@nithinraok
Copy link
Member

That failing CI test is okay to skip as its flaky. @chtruong814 FYI

Copy link
Member

@nithinraok nithinraok left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

@chtruong814 / @ko3n1g review for docker related changes.

@blisc blisc merged commit 737f9d5 into NVIDIA-NeMo:main Jan 16, 2026
259 checks passed
@blisc blisc deleted the tts_2512_removetorchaudio branch January 16, 2026 15:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants