Michael Wyatt
|
309aaa459f
Update tests/unit/inference/test_inference.py
|
1 year ago |
Michael Wyatt
|
353a4c10fd
Update tests/unit/inference/test_inference.py
|
1 year ago |
Logan Adams
|
8552bd6a05
Merge branch 'master' into loadams/low-cpu-mem-ut
|
1 year ago |
Michael Wyatt
|
d9a889d559
Fix nv-nightly workflow (#4163)
|
1 year ago |
Logan Adams
|
d1bc2cb8ca
Format and PR feedback means that we can remove the import we added.
|
1 year ago |
Logan Adams
|
c26b7c73b2
Whitespace and PR feedback
|
1 year ago |
Logan Adams
|
61e6d06965
Merge branch 'master' into loadams/low-cpu-mem-ut
|
1 year ago |
Wang, Yi
|
5e16eb2c93
enable autoTP for mpt in huggingface model hub without trust_remote_code (#4062)
|
1 year ago |
Logan Adams
|
8fb111c08d
Treat empty environment variables as unset in (#4185)
|
1 year ago |
Logan Adams
|
c5c3093bf0
Merge branch 'master' into loadams/low-cpu-mem-ut
|
1 year ago |
Logan Adams
|
4746266a15
Remove invalid model test config as that was removed.
|
1 year ago |
Logan Adams
|
f19a89a769
Accidently dropped pytestmark from other test
|
1 year ago |
Logan Adams
|
ae7166af38
final format fix
|
1 year ago |
Logan Adams
|
1edb452197
Fix chanves
|
1 year ago |
Logan Adams
|
1dfc7ff040
Indentation fix
|
1 year ago |
mzl
|
7f3e82fe09
do allgather only in shared optimizer states groups (#4167)
|
1 year ago |
Logan Adams
|
6c1ec0961a
Formatting has changes
|
1 year ago |
Logan Adams
|
00f47f033a
Merge branch 'master' into lokoppak/low_cpu_mem_usage_ut
|
1 year ago |
Heyang Qin
|
7711bdbbd2
MP ZeRO++ (#3954)
|
1 year ago |
Jeff Rasley
|
f036f00c37
bump to 0.10.2
|
1 year ago |
Michael Wyatt
|
46d859a75d
pin transformers to last known good commit (#4174)
|
1 year ago |
Lev Kurilenko
|
a3540f17f8
Add DSE branch input to nv-ds-chat (#4173)
|
1 year ago |
Ma, Guokai
|
19e9a7c028
[CPU][Bugfix] Make uid and addr_port part of SHM name in CCL backend (#4115)
|
1 year ago |
Lev Kurilenko
|
64c670ef02
Add DS-Chat CI workflow (#4127)
|
1 year ago |
Michael Wyatt
|
bd65eeaf70
fix badges (#4162)
|
1 year ago |
Logan Adams
|
1a29573946
Handling for SIGTERM as well (#4160)
|
1 year ago |
Sam Foreman
|
740b78050d
Fixes #4151 (#4152)
|
1 year ago |
Molly Smith
|
341cefd2a4
Return nn.parameter type for weights and biases (#4146)
|
1 year ago |
Logan Adams
|
a452301879
Remove incorrect async-io library checking code. (#4150)
|
1 year ago |
Olatunji Ruwase
|
9d79cfd1e9
Respect memory pinning config (#4131)
|
1 year ago |