NVIDIA / NeMo-Aligner

Scalable toolkit for efficient model alignment
Apache License 2.0
628 stars 78 forks source link

Fix `dev` branch's build after PTL upgrade #418

Open terrykong opened 5 days ago

terrykong commented 5 days ago

Describe the bug

We had to force merge https://github.com/NVIDIA/NeMo-Aligner/pull/410 because internal CI was failing even though GH presubmit was failing because of a build issue on the public dockerfile/environment.

Steps/Code to reproduce bug

Please list minimal steps or code snippet for us to be able to reproduce the bug.

A helpful guide on on how to craft a minimal bug report http://matthewrocklin.com/blog/work/2018/02/28/minimal-bug-reports.

Expected behavior

A clear and concise description of what you expected to happen.

Environment overview (please complete the following information)

Environment details

If NVIDIA docker image is used you don't need to specify these. Otherwise, please provide:

Additional context

Add any other context about the problem here. Example: GPU model