Update README.md
Browse files
README.md
CHANGED
@@ -23,15 +23,20 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
23 |
)
|
24 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
25 |
|
26 |
-
input_ids = tokenizer("
|
27 |
-
outputs = model.generate(input_ids, max_new_tokens=
|
28 |
print(tokenizer.batch_decode(outputs)[0])
|
29 |
"""
|
30 |
-
<s>
|
31 |
|
32 |
-
|
|
|
|
|
|
|
|
|
33 |
"""
|
34 |
```
|
|
|
35 |
|
36 |
### Parameter count
|
37 |
|
|
|
23 |
)
|
24 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
25 |
|
26 |
+
input_ids = tokenizer("In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English.", return_tensors="pt").input_ids.cuda()
|
27 |
+
outputs = model.generate(input_ids, max_new_tokens=128, temperature=0.7, do_sample=True, top_p=0.95, repetition_penalty=1.1)
|
28 |
print(tokenizer.batch_decode(outputs)[0])
|
29 |
"""
|
30 |
+
<s> In a shocking finding, scientist discovered a herd of unicorns living in a remote, previously unexplored valley, in the Andes Mountains. Even more surprising to the researchers was the fact that the unicorns spoke perfect English. It turns out that only a few people believed their father’s name had been made public by their mother and her father.
|
31 |
|
32 |
+
As the study finds, scientists discovered the secretive organic matter in the Andes mountains: “In a forest surrounded by a stream of lakes, an unseen rock formed with a series of tuberous orbits.”
|
33 |
+
|
34 |
+
They found that the mysterious bodies were buried in some parts of the mountain, known as the Andes mountain. The researchers then searched for the body, which they identified with the Butterfly.
|
35 |
+
|
36 |
+
The discovery of the body is the result of
|
37 |
"""
|
38 |
```
|
39 |
+
(it's... a little undertrained! thats okay!)
|
40 |
|
41 |
### Parameter count
|
42 |
|