PingVortex commited on
Commit
fbe96e9
·
verified ·
1 Parent(s): 976f390

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "activation_function": "gelu_new",
3
  "architectures": [
4
- "GPT2LMHeadModel"
5
  ],
6
  "attn_pdrop": 0.1,
7
  "bos_token_id": 50256,
@@ -30,7 +30,7 @@
30
  "summary_proj_to_labels": true,
31
  "summary_type": "cls_index",
32
  "summary_use_proj": true,
33
- "torch_dtype": "float32",
34
  "transformers_version": "4.51.3",
35
  "use_cache": true,
36
  "vocab_size": 50258
 
1
  {
2
  "activation_function": "gelu_new",
3
  "architectures": [
4
+ "GPT2Model"
5
  ],
6
  "attn_pdrop": 0.1,
7
  "bos_token_id": 50256,
 
30
  "summary_proj_to_labels": true,
31
  "summary_type": "cls_index",
32
  "summary_use_proj": true,
33
+ "torch_dtype": "float16",
34
  "transformers_version": "4.51.3",
35
  "use_cache": true,
36
  "vocab_size": 50258