Thanks your great work!
#1
by
bay-llm
- opened
thanks your great work!
I want to know how to use reward model too.
Best regards!
Hi @bay-llm I tried exporting the Reward model and it failed because there were missing tensors in the Nemo file. I used the checkpoint from here https://catalog.ngc.nvidia.com/orgs/nvidia/teams/nemo/models/nemotron-4-340b-reward
Thank you for your response. I am hopeful that this will resolve the issue about the reward model weights.
bay-llm
changed discussion status to
closed