minhtoan commited on
Commit
41e3f63
·
1 Parent(s): 7f35677

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -5
README.md CHANGED
@@ -20,7 +20,6 @@ Pretrained gpt model on Vietnamese New
20
  # How to use the model
21
 
22
  ~~~~
23
- import torch
24
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
25
 
26
  tokenizer = GPT2Tokenizer.from_pretrained('minhtoan/gpt2-finetune-vietnamese-news')
@@ -34,10 +33,6 @@ sample_outputs = model.generate(input_ids,pad_token_id=tokenizer.eos_token_id,
34
  do_sample=True,
35
  max_length=max_length,
36
  min_length=max_length,
37
- top_k=40,
38
- num_beams=5,
39
- early_stopping=True,
40
- no_repeat_ngram_size=2,
41
  num_return_sequences=3)
42
 
43
  for i, sample_output in enumerate(sample_outputs):
 
20
  # How to use the model
21
 
22
  ~~~~
 
23
  from transformers import GPT2Tokenizer, GPT2LMHeadModel
24
 
25
  tokenizer = GPT2Tokenizer.from_pretrained('minhtoan/gpt2-finetune-vietnamese-news')
 
33
  do_sample=True,
34
  max_length=max_length,
35
  min_length=max_length,
 
 
 
 
36
  num_return_sequences=3)
37
 
38
  for i, sample_output in enumerate(sample_outputs):