Update README.md
Browse files
README.md
CHANGED
@@ -60,6 +60,12 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
|
|
60 |
|
61 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/lts-gpt2-sm")
|
62 |
model = AutoModelForCausalLM.from_pretrained("microsoft/lts-gpt2-sm", subfolder="gpt2_0e1b5a3c867d6473da270799061f3089a1df5afd")
|
|
|
|
|
|
|
|
|
|
|
|
|
63 |
```
|
64 |
|
65 |
## Training Details
|
|
|
60 |
|
61 |
tokenizer = AutoTokenizer.from_pretrained("microsoft/lts-gpt2-sm")
|
62 |
model = AutoModelForCausalLM.from_pretrained("microsoft/lts-gpt2-sm", subfolder="gpt2_0e1b5a3c867d6473da270799061f3089a1df5afd")
|
63 |
+
|
64 |
+
text = "# Halo Infinite Review"
|
65 |
+
input_ids = tokenizer(text, return_tensors="pt").input_ids
|
66 |
+
|
67 |
+
generated_ids = model.generate(input_ids, max_length=128)
|
68 |
+
print(tokenizer.decode(generated_ids[0], skip_special_tokens=True))
|
69 |
```
|
70 |
|
71 |
## Training Details
|