ailments commited on
Commit
b4fc9ad
1 Parent(s): 2e28d5f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -0
config.json CHANGED
@@ -50,6 +50,17 @@
50
  "torch_dtype": "float32",
51
  "transformers_version": "4.35.0",
52
  "use_cache": true,
 
 
 
 
 
 
 
 
 
 
 
53
  "vocab_size": 50257,
54
  "window_size": 256
55
  }
 
50
  "torch_dtype": "float32",
51
  "transformers_version": "4.35.0",
52
  "use_cache": true,
53
+ "task_specific_params": {
54
+ "text-generation": {
55
+ "do_sample": true,
56
+ "temperature": 0.75,
57
+ "max_length": 100,
58
+ "top_p": 0.95,
59
+ "top_k": 50,
60
+ "wait_for_model": true,
61
+ "num_return_sequences": 2
62
+ }
63
+ },
64
  "vocab_size": 50257,
65
  "window_size": 256
66
  }