mihaimasala commited on
Commit
391be02
1 Parent(s): 0c0b737

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "Write me a poem about Machine Learning."
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