rudrashah commited on
Commit
f24afbf
1 Parent(s): 5eb36f0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -0
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
+ ```