Text Generation
Transformers
PyTorch
Safetensors
English
gpt_refact
code
custom_code
Eval Results
4 papers
svakhreev commited on
Commit
bdd25a3
1 Parent(s): 49e21c1

float16 as default type

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -24,7 +24,7 @@
24
  "resid_pdrop": 0.1,
25
  "scale_attention_softmax_in_fp32": false,
26
  "scale_attn_weights": true,
27
- "torch_dtype": "bfloat16",
28
  "transformers_version": "4.31.0",
29
  "use_cache": true,
30
  "vocab_size": 49216
 
24
  "resid_pdrop": 0.1,
25
  "scale_attention_softmax_in_fp32": false,
26
  "scale_attn_weights": true,
27
+ "torch_dtype": "float16",
28
  "transformers_version": "4.31.0",
29
  "use_cache": true,
30
  "vocab_size": 49216