Views
No views yet
1device = torch.device("cuda")
2REWARD_MODEL_REPO_NAME = f"MurDanya/llm-course-hw2-reward-model"
3
4tokenizer = AutoTokenizer.from_pretrained(REWARD_MODEL_REPO_NAME)
5reward_model = AutoModelForSequenceClassification.from_pretrained(REWARD_MODEL_REPO_NAME)