Hugging Face's logo

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

Text Classification
Transformers
Safetensors
mistral
custom_code
text-generation-inference
Model card Files
xet
Community
Haoxiang-Wang commited on
Commit
9e6cd97
·
verified ·
1 Parent(s): a179a64

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ Here is a sample code that you can try
42
  from transformers import AutoModelForSequenceClassification,AutoTokenizer
43
  import torch
44
  device = 'cuda'
45
- path = "Haoxiang-Wang/RewardModel-Mistral-7B-for-DPA-v1"
46
  rm = AutoModelForSequenceClassification.from_pretrained(path, trust_remote_code=True).to(device)
47
  tokenizer = AutoTokenizer.from_pretrained(path)
48
 
 
42
  from transformers import AutoModelForSequenceClassification,AutoTokenizer
43
  import torch
44
  device = 'cuda'
45
+ path = "RLHFlow/RewardModel-Mistral-7B-for-DPA-v1"
46
  rm = AutoModelForSequenceClassification.from_pretrained(path, trust_remote_code=True).to(device)
47
  tokenizer = AutoTokenizer.from_pretrained(path)
48