Update README.md

README.md

@@ -51,7 +51,7 @@ This repository has 21 checkpoints, saved as revisions, that were logged during
 
 ## Details
 
-- **Size:** 162 million parameters
+- **Size:** 162,417,408 million parameters
 - **Dataset:** [Portuguese-Corpus-v3](https://huggingface.co/datasets/nicholasKluge/portuguese-corpus-v3)
 - **Language:** Portuguese
 - **Number of steps:** 457,969
@@ -101,16 +101,31 @@ This repository has the [source code](https://github.com/Nkluge-correa/Aira) use
 ## Usage
 
 ```python
-# Use a pipeline as a high-level helper
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
 
-pipe = pipeline("text-generation", model="nicholasKluge/Teeny-tiny-llama-162m")
+# Load model and the tokenizer
+tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/Teeny-tiny-llama-162m", revision='main')
+model = AutoModelForCausalLM.from_pretrained("nicholasKluge/Teeny-tiny-llama-162m", revision='main')
 
-# Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
+# Pass the model to your device
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+model.eval()
+model.to(device)
+
+# Tokenize the inputs and pass them to the device
+inputs = tokenizer("Astronomia é a ciência", return_tensors="pt").to(device)
+
+# Generate some text
+completions = model.generate(**inputs, num_return_sequences=2, max_new_tokens=100)
+
+# Print the generated text
+for i, completion in enumerate(completions):
+    print(f'🤖 {tokenizer.decode(completion)}')
+
+>>> 🤖 <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.
 
-tokenizer = AutoTokenizer.from_pretrained("nicholasKluge/Teeny-tiny-llama-162m")
-model = AutoModelForCausalLM.from_pretrained("nicholasKluge/Teeny-tiny-llama-162m")
 ```
 
 ## Limitations