sdhan commited on
Commit
9ec4133
1 Parent(s): 1f6f0e3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -3
README.md CHANGED
@@ -1,3 +1,21 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ Model Details
6
+
7
+ dataset : DPO dataset (huggingface datasets, 2digit datasets 활용)
8
+
9
+ Training Method Method : DPO.
10
+
11
+ Usage
12
+ from transformers import AutoModelForCausalLM, AutoTokenizer
13
+ import torch
14
+ repo = "sdhan/SD_SOLAR_10.7B_v1.0"
15
+ model = AutoModelForCausalLM.from_pretrained(
16
+ repo,
17
+ return_dict=True,
18
+ torch_dtype=torch.float16,
19
+ device_map='auto'
20
+ )
21
+ tokenizer = AutoTokenizer.from_pretrained(repo)