NVIDIA-NeMo / NVIDIA-NeMo/RL

Integration between NeMo and NeMo-RL for audio LLMs

Open
#619 8 comments 0 reactions 0 assignees View on GitHub
community-request external needs-design waiting-on-customer x-tii
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

Hi, we are training our audio LLMs using the code provided in https://github.com/tiiuae/aiccu-audio-nemo/tree/main/nemo/collections/speechlm . Under the hood these are just MegatronLM models.

Now we'd like to apply RL training to these models and we'd like to use NeMo-RL for that. What's the easiest way to load our models into NeMo-RL? Is this something that's relatively simple to add or do you think it's a really big task?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.