AIdenU
/

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

Text Generation

text-generation-inference

Model card Files Files and versions

AIdenU commited on Dec 18, 2023

Commit

10c4f59

·

1 Parent(s): 3d338c9

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -8,6 +8,7 @@ tags:
 ### Model Generation
 from transforemrs import AutoTokenizer, AutoModelForCausalLM
 model = AutoModelForCausalLM.from_pretrained("AIdenU/LLAMA-2-13b-ko-Y24-DPO_v0.1", device_map="auto")
@@ -24,4 +25,5 @@ outputs = model.generate(
   top_p=1,
   do_sample=True
 )
-print(tokenizer.decode(outputs[0]))

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