nicholasKluge
commited on
Commit
•
2d6b65e
1
Parent(s):
4736d3c
Update README.md
Browse files
README.md
CHANGED
@@ -98,7 +98,27 @@ This repository has the [source code](https://github.com/Nkluge-correa/Aira) use
|
|
98 |
| | checkpointing_steps | 22000 |
|
99 |
| | tf32 | true |
|
100 |
|
101 |
-
##
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
102 |
|
103 |
```python
|
104 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
98 |
| | checkpointing_steps | 22000 |
|
99 |
| | tf32 | true |
|
100 |
|
101 |
+
## Basic usage
|
102 |
+
|
103 |
+
Using the `pipeline`:
|
104 |
+
|
105 |
+
```python
|
106 |
+
|
107 |
+
# Use a pipeline as a high-level helper
|
108 |
+
from transformers import pipeline
|
109 |
+
|
110 |
+
generator = pipeline("text-generation", model="nicholasKluge/Teeny-tiny-llama-162m")
|
111 |
+
|
112 |
+
completions = generator("Astronomia é a ciência", num_return_sequences=1, max_new_tokens=100)
|
113 |
+
|
114 |
+
for comp in completions:
|
115 |
+
print(f"🤖 {comp['generated_text']}")
|
116 |
+
|
117 |
+
>>> 🤖 <s> Astronomia é a ciência que estuda o universo e as leis da física e suas relações com os fenômenos naturais e seus efeitos sobre o meio ambiente e o homem. A astronomia é uma disciplina científica que se dedica à investigação de fenômenos astronômicos e ao estudo das propriedades dos objetos celestes.
|
118 |
+
|
119 |
+
```
|
120 |
+
|
121 |
+
Using the `AutoTokenizer` and `AutoModelForCausalLM`:
|
122 |
|
123 |
```python
|
124 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|