To download run
from transformers import AutoModel model = AutoModel.from_pretrained("LawInformedAI/gpt-j-reward-model", trust_remote_code=True)