Update README.md

README.md

@@ -1,5 +1,18 @@
+To use this model, you need to load by `AutoModelForSequenceClassification`,
+```python
+model = AutoModelForSequenceClassification.from_pretrained(
+    "hendrydong/Mistral-RM-for-RAFT-GSHF-v0", num_labels=1, torch_dtype=torch.bfloat16
+)
+```
+and prepare dataset like
+```python
+SAMPLE =[
+{'role': 'user', 'content': 'Hi!'},
+{'role': 'assistant', 'content': 'How are you?'},
+]
+```
 
-
+The template is the same as `mistralai/Mistral-7B-Instruct-v0.2`.
 
 The reward model can be used for iterative SFT/DPO