update config.json
Browse filesSigned-off-by: Tang, Kaihui <kaihui.tang@intel.com>
- config.json +3 -2
config.json
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
{
|
|
|
|
| 2 |
"activation_function": "gelu_new",
|
| 3 |
"architectures": [
|
| 4 |
"GPTJForCausalLM"
|
|
@@ -25,7 +26,6 @@
|
|
| 25 |
"summary_proj_to_labels": true,
|
| 26 |
"summary_type": "cls_index",
|
| 27 |
"summary_use_proj": true,
|
| 28 |
-
"torchscript": true,
|
| 29 |
"task_specific_params": {
|
| 30 |
"text-generation": {
|
| 31 |
"do_sample": true,
|
|
@@ -35,7 +35,8 @@
|
|
| 35 |
},
|
| 36 |
"tie_word_embeddings": false,
|
| 37 |
"tokenizer_class": "GPT2Tokenizer",
|
| 38 |
-
"
|
|
|
|
| 39 |
"use_cache": true,
|
| 40 |
"vocab_size": 50400
|
| 41 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"_name_or_path": "EleutherAI/gpt-j-6b",
|
| 3 |
"activation_function": "gelu_new",
|
| 4 |
"architectures": [
|
| 5 |
"GPTJForCausalLM"
|
|
|
|
| 26 |
"summary_proj_to_labels": true,
|
| 27 |
"summary_type": "cls_index",
|
| 28 |
"summary_use_proj": true,
|
|
|
|
| 29 |
"task_specific_params": {
|
| 30 |
"text-generation": {
|
| 31 |
"do_sample": true,
|
|
|
|
| 35 |
},
|
| 36 |
"tie_word_embeddings": false,
|
| 37 |
"tokenizer_class": "GPT2Tokenizer",
|
| 38 |
+
"torchscript": true,
|
| 39 |
+
"transformers_version": "4.32.0",
|
| 40 |
"use_cache": true,
|
| 41 |
"vocab_size": 50400
|
| 42 |
}
|