pearsonkyle commited on
Commit
76cf2ba
1 Parent(s): 7185fea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -23,7 +23,8 @@ llm = pipeline('text-generation',model='pearsonkyle/gpt2-arxiv', tokenizer=token
23
 
24
  texts = llm("Directly imaged exoplanets probe",
25
  max_length=50, do_sample=True, num_return_sequences=5,
26
- penalty_alpha=0.65, top_k=4, repetition_penalty=1.2)
 
27
 
28
  for i in range(5):
29
  print(texts[i]['generated_text']+'\n')
 
23
 
24
  texts = llm("Directly imaged exoplanets probe",
25
  max_length=50, do_sample=True, num_return_sequences=5,
26
+ penalty_alpha=0.65, top_k=40, repetition_penalty=1.25,
27
+ temperature=0.95)
28
 
29
  for i in range(5):
30
  print(texts[i]['generated_text']+'\n')