Lifan-Z commited on
Commit
e0e03c1
1 Parent(s): 5fd1fdf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -13,7 +13,7 @@ import transformers
13
  from transformers import pipeline
14
 
15
  gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
16
- sequences = gpt2('<|endoftext|>雨', max_length=50, num_return_sequences=6, eos_token_id=0)
17
  for seq in sequences:
18
  print(seq)
19
 
 
13
  from transformers import pipeline
14
 
15
  gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
16
+ sequences = gpt2('<|endoftext|>雨', max_length=50, do_sample=True, top_k=20, top_p=0.9, num_return_sequences=6, eos_token_id=0)
17
  for seq in sequences:
18
  print(seq)
19