Home Original page

cp: Explicitly set torch >= 2.6.0 and remove weights_only=False (#15314) to r2.6.0 by chtruong814 · Pull Request #15337 · NVIDIA-NeMo/NeMo

@chtruong814 

…15314)

* Set torch minimum version to be 2.6.0

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Update required torch version to >= 2.6 in docs

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Remove references to weights_only=False

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Apply isort and black reformatting

Signed-off-by: chtruong814 <chtruong814@users.noreply.github.com>

* Set TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1 during tests

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Fix tests

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Add TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1 to speech tests

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

* Update required pytorch version in docs to 2.6+

Signed-off-by: Charlie Truong <chtruong@nvidia.com>

---------

Signed-off-by: Charlie Truong <chtruong@nvidia.com>
Signed-off-by: chtruong814 <chtruong814@users.noreply.github.com>
Co-authored-by: chtruong814 <chtruong814@users.noreply.github.com>
Signed-off-by: Charlie Truong <chtruong@nvidia.com>