SreyanG-NVIDIA commited on
Commit
ac8bd83
·
verified ·
1 Parent(s): 65d1596

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -0
requirements.txt CHANGED
@@ -39,6 +39,9 @@ pre-commit
39
  loguru
40
  hydra-core
41
  xgrammar
 
 
 
42
 
43
  # Direct Git dependency
44
  git+https://github.com/bfshi/scaling_on_scales#egg=s2wrapper
 
39
  loguru
40
  hydra-core
41
  xgrammar
42
+ deepspeed==0.9.5
43
+ wandb
44
+ ninja
45
 
46
  # Direct Git dependency
47
  git+https://github.com/bfshi/scaling_on_scales#egg=s2wrapper