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

Merged

Commits

Pin torch in nv-ds-chat workflow

loadams committed 1 year ago
Turn off NCCL debug

loadams committed 1 year ago
Ensure we get torch+cu vs torch+cpu

loadams committed 1 year ago
Merge branch 'master' into loadams/ds-chat-fixes

loadams committed 1 year ago
Add torchvision at same install time as torch

loadams committed 1 year ago

Loading