sdhan
/

SD_SOLAR_10.7B_v1.0

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

sdhan commited on May 12

Commit

a32d708

•

1 Parent(s): 9ec4133

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -9,6 +9,7 @@ dataset : DPO dataset (huggingface datasets, 2digit datasets 활용)
 Training Method Method : DPO.
 Usage
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 repo = "sdhan/SD_SOLAR_10.7B_v1.0"
@@ -18,4 +19,5 @@ model = AutoModelForCausalLM.from_pretrained(
         torch_dtype=torch.float16,
         device_map='auto'
 )
-tokenizer = AutoTokenizer.from_pretrained(repo)

 Training Method Method : DPO.
 Usage
+```
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 repo = "sdhan/SD_SOLAR_10.7B_v1.0"
         torch_dtype=torch.float16,
         device_map='auto'
 )
+tokenizer = AutoTokenizer.from_pretrained(repo)
+```