Update README.md
Browse files
README.md
CHANGED
|
@@ -55,7 +55,9 @@ model = AutoModelForCausalLM.from_pretrained(quantized_model_dir,
|
|
| 55 |
)
|
| 56 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, use_fast=True)
|
| 57 |
print(tokenizer.decode(model.generate(**tokenizer("There is a girl who likes adventure,", return_tensors="pt").to(model.device),max_new_tokens=50)[0]))
|
| 58 |
-
|
|
|
|
|
|
|
| 59 |
```
|
| 60 |
|
| 61 |
|
|
|
|
| 55 |
)
|
| 56 |
tokenizer = AutoTokenizer.from_pretrained(quantized_model_dir, use_fast=True)
|
| 57 |
print(tokenizer.decode(model.generate(**tokenizer("There is a girl who likes adventure,", return_tensors="pt").to(model.device),max_new_tokens=50)[0]))
|
| 58 |
+
"""
|
| 59 |
+
<s> There is a girl who likes adventure, who loves to travel, who is always looking for new experiences. She is a dreamer, a doer, a thinker, a believer. She is a girl who is not afraid to take risks, to make mistakes, to learn from
|
| 60 |
+
"""
|
| 61 |
```
|
| 62 |
|
| 63 |
|