Lifan-Z commited on
Commit
0a969d4
1 Parent(s): 5139cfd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -6
README.md CHANGED
@@ -11,17 +11,17 @@ import transformers
11
  from transformers import pipeline
12
 
13
  gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
14
- sequences = gpt2('<|endoftext|>雨', max_length=50, do_sample=True, top_k=950, repetition_penalty=1.2, num_return_sequences=6, eos_token_id=0)
15
  for seq in sequences:
16
  print(seq)
17
 
18
  Output:
19
- {'generated_text': '<|endoftext|>雨 过 鸿 。'}
20
- {'generated_text': '<|endoftext|>雨 后 。'}
21
  {'generated_text': '<|endoftext|>雨 余 天 影 半 , 路 僻 出 阶 除 。 树 矗 山 扉 重 , 溪 泠 水 汽 清 。 耕 夫 齐 舍 逐 , 兵 卒 峻 庭 隆 。 何 必 投 吾 圃 , 寻 根 傍 石 林 。'}
22
- {'generated_text': '<|endoftext|>雨 绿 。'}
23
- {'generated_text': '<|endoftext|>雨 线 鸿 , 与 。'}
24
- {'generated_text': '<|endoftext|>雨 声 。 。'}
25
  ```
26
 
27
  ### **References:**
 
11
  from transformers import pipeline
12
 
13
  gpt2 = pipeline('text-generation', model = "Lifan-Z/Chinese-Classic-Poem-Generator-style5x8-GPT2")
14
+ sequences = gpt2('<|endoftext|>雨', max_length=50, do_sample=True, top_k=20, top_p=0.9, repetition_penalty=1.2, num_return_sequences=6, eos_token_id=0)
15
  for seq in sequences:
16
  print(seq)
17
 
18
  Output:
19
+ {'generated_text': '<|endoftext|>雨 过 。'}
20
+ {'generated_text': '<|endoftext|>雨 后 。'}
21
  {'generated_text': '<|endoftext|>雨 余 天 影 半 , 路 僻 出 阶 除 。 树 矗 山 扉 重 , 溪 泠 水 汽 清 。 耕 夫 齐 舍 逐 , 兵 卒 峻 庭 隆 。 何 必 投 吾 圃 , 寻 根 傍 石 林 。'}
22
+ {'generated_text': '<|endoftext|>雨 。'}
23
+ {'generated_text': '<|endoftext|>雨 。'}
24
+ {'generated_text': '<|endoftext|>雨 穿 。'}
25
  ```
26
 
27
  ### **References:**