RLHFlow
/

RewardModel-Mistral-7B-for-DPA-v1

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Haoxiang-Wang commited on May 9

Commit

9e6cd97

•

1 Parent(s): a179a64

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -42,7 +42,7 @@ Here is a sample code that you can try
 from transformers import AutoModelForSequenceClassification,AutoTokenizer
 import torch
 device = 'cuda'
-path = "Haoxiang-Wang/RewardModel-Mistral-7B-for-DPA-v1"
 rm = AutoModelForSequenceClassification.from_pretrained(path, trust_remote_code=True).to(device)
 tokenizer = AutoTokenizer.from_pretrained(path)

 from transformers import AutoModelForSequenceClassification,AutoTokenizer
 import torch
 device = 'cuda'
+path = "RLHFlow/RewardModel-Mistral-7B-for-DPA-v1"
 rm = AutoModelForSequenceClassification.from_pretrained(path, trust_remote_code=True).to(device)
 tokenizer = AutoTokenizer.from_pretrained(path)