Tags: DavitGrigoryan132/NeMo
Tags
chore: Add changelog (NVIDIA#11283) * chore: Update CHANGELOG Signed-off-by: Oliver Koenig <okoenig@nvidia.com> * add highlights Signed-off-by: Oliver Koenig <okoenig@nvidia.com> * hyperlinks Signed-off-by: Oliver Koenig <okoenig@nvidia.com> * links Signed-off-by: Oliver Koenig <okoenig@nvidia.com> --------- Signed-off-by: Oliver Koenig <okoenig@nvidia.com>
Support TE-DPA For Stable Diffusion (NVIDIA#10314) * [SD] Add te-dpa support Signed-off-by: Wil Kong <alpha0422@gmail.com> * [SD] Add te-dpa support, resolve compatibility with TE-master Signed-off-by: Wil Kong <alpha0422@gmail.com> * [SD] Add te-dpa support, add check for attention configs. Signed-off-by: Wil Kong <alpha0422@gmail.com> * Fix bugs of flash-attn and dpa in SD. Signed-off-by: Wil Kong <alpha0422@gmail.com> * Fix the issue of DPA API change. Signed-off-by: Wil Kong <alpha0422@gmail.com> * Apply isort and black reformatting Signed-off-by: alpha0422 <alpha0422@users.noreply.github.com> Signed-off-by: Wil Kong <alpha0422@gmail.com> * [SD] TE-DPA: disbale use te-dpa in inference flow. --------- Signed-off-by: Wil Kong <alpha0422@gmail.com> Signed-off-by: alpha0422 <alpha0422@users.noreply.github.com> Co-authored-by: Mengdi Wang <didow@nvidia.com>
add manifest file (NVIDIA#10161) Signed-off-by: Oliver Koenig <okoenig@nvidia.com>
Add option for mutex timeout in distributed optimizer backward hook (N… …VIDIA#9087) * Tim: Add option for timeout in distopt callback mutex Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> * Replace parent's _lock Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> * Revert "Replace parent's _lock" This reverts commit 972d1b6. Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> * Raise RuntimeError when timeout Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> * Change RuntimeError to print Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> --------- Signed-off-by: Jaemin Choi <jaeminc@nvidia.com> Co-authored-by: Jaemin Choi <jaeminc@nvidia.com>
update github raw content link (NVIDIA#8517) Signed-off-by: Chen Cui <chcui@nvidia.com>
Update Apex install command in Dockerfile (NVIDIA#7794) * move core install to /workspace (NVIDIA#7706) Signed-off-by: Abhinav Khattar <aklife97@gmail.com> * update apex install in dockerfile Signed-off-by: eharper <eharper@nvidia.com> * use fetch head Signed-off-by: eharper <eharper@nvidia.com> --------- Signed-off-by: Abhinav Khattar <aklife97@gmail.com> Signed-off-by: eharper <eharper@nvidia.com> Co-authored-by: Abhinav Khattar <aklife97@gmail.com>
PreviousNext