Update README.md

README.md

@@ -1,10 +1,10 @@
 ---
 license: apache-2.0
 tags:
-- merge
 - mergekit
 - EmbeddedLLM/Mistral-7B-Merge-14-v0.1
 - OpenPipe/mistral-ft-optimized-1218
+- NLP
 ---
 
 # RLM-mini
@@ -17,6 +17,7 @@ RLM-mini is a 7.2 Billion parameter model,RLM-mini is designed to provide a robu
 
 # Usage
 
+### Direct Model
 ``` python
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
@@ -28,3 +29,28 @@ output = model.generate(**input_token, max_length=250)
 output = tokenizer.decode(output[0])
 
 ```
+
+### Using Pipeline
+``` python
+from transformers import AutoTokenizer
+import transformers
+import torch
+
+model = "rudrashah/RLM-mini"
+messages = [{"role": "user", "content": "What is a large language model?"}]
+
+tokenizer = AutoTokenizer.from_pretrained(model)
+prompt = tokenizer.apply_chat_template(
+    messages,
+    tokenize=False,
+    add_generation_prompt=True
+)
+pipeline = transformers.pipeline(
+    "text-generation",
+    model=model,
+    torch_dtype=torch.float16,
+    device_map="auto",
+)
+
+outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+```