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