Update README.md
Browse files
README.md
CHANGED
@@ -17,3 +17,15 @@ RLM-mini is a 7.2 Billion parameter model,RLM-mini is designed to provide a robu
|
|
17 |
|
18 |
# Usage
|
19 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
# Usage
|
19 |
|
20 |
+
``` python
|
21 |
+
|
22 |
+
from transformers import AutoTokenizer, AutoModelForCausalLM
|
23 |
+
|
24 |
+
tokenizer = AutoTokenizer.from_pretrained("rudrashah/RLM-mini")
|
25 |
+
model = AutoModelForCausalLM.from_pretrained("rudrashah/RLM-mini")
|
26 |
+
|
27 |
+
input_token = tokenizer("How to make Pav Bhaji?", return_tensors="pt")
|
28 |
+
output = model.generate(**input_token, max_length=250)
|
29 |
+
output = tokenizer.decode(output[0])
|
30 |
+
|
31 |
+
```
|