joanrodai commited on
Commit
a84759b
·
verified ·
1 Parent(s): 1780344

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -22,7 +22,7 @@
22
  "num_hidden_layers": 24,
23
  "num_kv_heads": 4,
24
  "starcoder_model_name": "bigcode/starcoderbase-1b",
25
- "torch_dtype": "float32",
26
  "train_LLM": true,
27
  "train_image_encoder": false,
28
  "transformers_version": "4.40.1",
 
22
  "num_hidden_layers": 24,
23
  "num_kv_heads": 4,
24
  "starcoder_model_name": "bigcode/starcoderbase-1b",
25
+ "torch_dtype": "float16",
26
  "train_LLM": true,
27
  "train_image_encoder": false,
28
  "transformers_version": "4.40.1",