gugarosa commited on
Commit
8d5bdf1
·
1 Parent(s): f971c45

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -0
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