Ray2333
/

GRM-gemma2-2B-rewardmodel-ft

Text Classification

Model card Files Files and versions Community

Ray2333 commited on 23 days ago

Commit

7183391

•

1 Parent(s): 4f7d3b8

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 device = 'cuda:0'
 # load model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained('Ray2333/GRM-Gemma-2B-rewardmodel-ft')
 reward_model = AutoModelForSequenceClassification.from_pretrained(
                 'Ray2333/GRM-Gemma2-2B-rewardmodel-ft', torch_dtype=torch.float16,
                 device_map=device,

 device = 'cuda:0'
 # load model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained('Ray2333/GRM-Gemma2-2B-rewardmodel-ft')
 reward_model = AutoModelForSequenceClassification.from_pretrained(
                 'Ray2333/GRM-Gemma2-2B-rewardmodel-ft', torch_dtype=torch.float16,
                 device_map=device,