DeepSpeed
3388f833 - Update container version that runs on A6000 tests. (#7153)

Comment changes are shownComment changes are hidden
Commit
87 days ago
Update container version that runs on A6000 tests. (#7153) Changes from https://github.com/huggingface/transformers/pull/36654 in transformers cause issues with the torch 2.5 version we were using. This just updated us to use a newer version. --------- Signed-off-by: Logan Adams <loadams@microsoft.com>
Author
Parents
  • .github/workflows
    • File
      nv-a6000.yml
    • File
      nv-flash-attn.yml
    • File
      nv-human-eval.yml