Update README.md

README.md

@@ -37,6 +37,27 @@ This is the model card of a 🤗 transformers model that has been pushed on the
 ## Uses
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+```python
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+model = AutoModelForCausalLM.from_pretrained("thangvip/vilord-1.8B-instruct", device_map="auto", cache_dir="./cache").eval()
+
+tokenizer = AutoTokenizer.from_pretrained("thangvip/vilord-1.8B-instruct", cache_dir="./cache")
+
+
+messages = [
+  {'role': 'system', 'content': "bạn là trợ lý AI hữu ích"},
+  {"role": "user", "content": "Nước nào có diện tích lớn nhất?"},
+]
+text = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
+print(text)
+inputs = tokenizer(text, return_tensors="pt")
+inputs = {k: v.to("cuda") for k, v in inputs.items()}
+
+outputs = model.generate(**inputs, tokenizer=tokenizer, max_new_tokens=256, do_sample=True, top_p=0.95, temperature=0.1, repetition_penalty=1.2, eos_token_id=tokenizer.eos_token_id, stop_strings=['<|im_end|>'])
+print(tokenizer.decode(outputs[0]))
+```
 
 ### Direct Use