FredZhang7 commited on
Commit
7f3802c
1 Parent(s): 6baf443

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -3
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "distilgpt2",
3
  "_num_labels": 1,
4
  "activation_function": "gelu_new",
5
  "architectures": [
@@ -37,11 +37,14 @@
37
  "task_specific_params": {
38
  "text-generation": {
39
  "do_sample": true,
40
- "max_length": 50
 
 
 
41
  }
42
  },
43
  "torch_dtype": "float32",
44
- "transformers_version": "4.21.2",
45
  "use_cache": true,
46
  "vocab_size": 50257
47
  }
 
1
  {
2
+ "_name_or_path": "./config.json",
3
  "_num_labels": 1,
4
  "activation_function": "gelu_new",
5
  "architectures": [
 
37
  "task_specific_params": {
38
  "text-generation": {
39
  "do_sample": true,
40
+ "max_length": 80,
41
+ "temperature": 0.7,
42
+ "top_k": 4,
43
+ "early_stopping": true
44
  }
45
  },
46
  "torch_dtype": "float32",
47
+ "transformers_version": "4.25.1",
48
  "use_cache": true,
49
  "vocab_size": 50257
50
  }