DeepSpeed
c963c21f - Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6) (#6982)

Commit
312 days ago
Specify torchvision in nv-ds-chat workflow (prevents errors with torch 2.6) (#6982) Fixes #6984. The workflow was pulling the updated torch 2.6, which caused CI failures. This keeps us on torch 2.5 for now, since installing torchvision as a dependency later on was pulling torch 2.6 with it which was unintended. This PR also unsets NCCL_DEBUG to avoid a large print out in the case of any errors.
Author
Parents
Loading