PeterLauLukCh commited on
Commit
5c2f20c
·
verified ·
1 Parent(s): 69feab9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -4,4 +4,19 @@ datasets:
4
  - trl-lib/ultrafeedback_binarized
5
  base_model:
6
  - alignment-handbook/zephyr-7b-sft-full
7
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  - trl-lib/ultrafeedback_binarized
5
  base_model:
6
  - alignment-handbook/zephyr-7b-sft-full
7
+ ---
8
+
9
+ ## How to use from the 🤗 Transformers library
10
+
11
+ ```python
12
+ from transformers import pipeline
13
+
14
+ messages = [
15
+ {"role": "user", "content": "Who are you?"}
16
+ ]
17
+ pipe = pipeline("text-generation", model="PeterLauLukCh/Mistral7B-trl_UltraFeedback-DPO", trust_remote_code=True)
18
+ pipe(messages)
19
+
20
+ from transformers import AutoModelForCausalLM
21
+
22
+ model = AutoModelForCausalLM.from_pretrained("PeterLauLukCh/Mistral7B-trl_UltraFeedback-DPO", trust_remote_code=True)