Transformers How to use DBangshu/gemma_ensemble with Transformers:
# Load model directly
from transformers import AutoTokenizer, RewardHead
tokenizer = AutoTokenizer.from_pretrained("DBangshu/gemma_ensemble")
model = RewardHead.from_pretrained("DBangshu/gemma_ensemble")