Text Generation
Transformers
PyTorch
English
gpt_neo
ColleenMacklin commited on
Commit
d32a670
·
1 Parent(s): fc91012

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -1
config.json CHANGED
@@ -53,7 +53,12 @@
53
  "task_specific_params": {
54
  "text-generation": {
55
  "do_sample": true,
56
- "max_length": 100
 
 
 
 
 
57
  }
58
  },
59
  "vocab_size": 50257,
 
53
  "task_specific_params": {
54
  "text-generation": {
55
  "do_sample": true,
56
+ "temperature": 0.7,
57
+ "max_length": 100,
58
+ "top_p": 0.95,
59
+ "top_k": 0.50,
60
+ "wait_for_model": true,
61
+ "num_return_sequences": 2
62
  }
63
  },
64
  "vocab_size": 50257,