AIdenU
/

LLAMA-2-13b-ko-Y24-DPO_v2.1

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

AIdenU commited on Feb 19

Commit

c0f610c

•

1 Parent(s): 08bf8ba

Update README.md

Files changed (1) hide show

README.md +6 -1

README.md CHANGED Viewed

@@ -6,6 +6,10 @@ pipeline_tag: text-generation
 tags:
 - llama2
 ---
 from transforemrs import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("AIdenU/LLAMA-2-13b-ko-Y24-DPO_v2.1", device_map="auto")
@@ -23,4 +27,5 @@ outputs = model.generate(
   top_p=1,
   do_sample=True
 )
-print(tokenizer.decode(outputs[0]))

 tags:
 - llama2
 ---
+### Model Generation
+```
 from transforemrs import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("AIdenU/LLAMA-2-13b-ko-Y24-DPO_v2.1", device_map="auto")
   top_p=1,
   do_sample=True
 )
+print(tokenizer.decode(outputs[0]))
+```