ColleenMacklin commited on
Commit
81c2ba1
1 Parent(s): a1b7b2f

Update config.json

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