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!

Owner

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

Sign up or log in to comment