prompting.validators.reward.prompt
#
Module Contents#
Classes#
- class prompting.validators.reward.prompt.PromptRewardModel(device)#
Bases:
prompting.validators.reward.reward.BaseRewardModel
- Parameters:
device (str) –
- reward(prompt, completion, name)#
- Parameters:
- Return type:
- get_rewards(prompt, completions, name)#
- Parameters:
- Return type: