taeminlee commited on
Commit
37bab52
โ€ข
1 Parent(s): 529fb41

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -0
README.md CHANGED
@@ -16,6 +16,9 @@ KoGPT2 on Huggingface Transformers
16
  ```python
17
  from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
18
 
 
 
 
19
  input_ids = tokenizer.encode("์•ˆ๋…•", add_special_tokens=False, return_tensors="pt")
20
  output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=100, num_return_sequences=3)
21
  for generated_sequence in output_sequences:
16
  ```python
17
  from transformers import GPT2LMHeadModel, PreTrainedTokenizerFast
18
 
19
+ model = GPT2LMHeadModel.from_pretrained("taeminlee/kogpt2")
20
+ tokenizer = PreTrainedTokenizerFast.from_pretrained("taeminlee/kogpt2")
21
+
22
  input_ids = tokenizer.encode("์•ˆ๋…•", add_special_tokens=False, return_tensors="pt")
23
  output_sequences = model.generate(input_ids=input_ids, do_sample=True, max_length=100, num_return_sequences=3)
24
  for generated_sequence in output_sequences: