How to use Digsm003/safety-assign1-bert-reward-model with Transformers:
# Load model directly from transformers import AutoTokenizer, BertForReward tokenizer = AutoTokenizer.from_pretrained("Digsm003/safety-assign1-bert-reward-model") model = BertForReward.from_pretrained("Digsm003/safety-assign1-bert-reward-model")
The community tab is the place to discuss and collaborate with the HF community!
