Text Generation
Transformers
PyTorch
Safetensors
English
gpt_refact
code
custom_code
Eval Results
mityasmc commited on
Commit
49e21c1
1 Parent(s): 71199b4

bfloat16 as default dtype

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": "float32",
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": "bfloat16",
28
  "transformers_version": "4.31.0",
29
  "use_cache": true,
30
  "vocab_size": 49216