README.md

AnyRewardModel

Usage

from transformers import AutoModel, AutoProcessor

model = AutoModel.from_pretrained("PKU-Alignment/AnyRewardModel", trust_remote_code=True)
processor = AutoProcessor.from_pretrained("PKU-Alignment/AnyRewardModel", trust_remote_code=True)

Note:

If you encounter the following error:

ModuleNotFoundError: No module named 'torchvision.transforms.functional_tensor'

Please refer to guide at blog for detailed resolution steps.