mihaimasala
commited on
Commit
•
391be02
1
Parent(s):
0c0b737
Update README.md
Browse files
README.md
CHANGED
@@ -57,10 +57,10 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
57 |
tokenizer = AutoTokenizer.from_pretrained("OpenLLM-Ro/RoLlama2-7b-Base")
|
58 |
model = AutoModelForCausalLM.from_pretrained("OpenLLM-Ro/RoLlama2-7b-Base")
|
59 |
|
60 |
-
input_text = "
|
61 |
input_ids = tokenizer(input_text, return_tensors="pt")
|
62 |
|
63 |
-
outputs = model.generate(**input_ids)
|
64 |
print(tokenizer.decode(outputs[0]))
|
65 |
```
|
66 |
|
|
|
57 |
tokenizer = AutoTokenizer.from_pretrained("OpenLLM-Ro/RoLlama2-7b-Base")
|
58 |
model = AutoModelForCausalLM.from_pretrained("OpenLLM-Ro/RoLlama2-7b-Base")
|
59 |
|
60 |
+
input_text = "Mihai Eminescu a fost "
|
61 |
input_ids = tokenizer(input_text, return_tensors="pt")
|
62 |
|
63 |
+
outputs = model.generate(**input_ids, max_new_tokens=100)
|
64 |
print(tokenizer.decode(outputs[0]))
|
65 |
```
|
66 |
|