martinezmatias commited on
Commit
11448eb
·
verified ·
1 Parent(s): b9ac0e6

Update config.json

Browse files

According to the [model.safetensors](https://huggingface.co/microsoft/phi-1?show_file_info=model.safetensors) file from Phi-1 is fp16, but this file indicates float32.
This PR proposes to fix that value.

Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "rope_scaling": null,
24
  "rope_theta": 10000.0,
25
  "tie_word_embeddings": false,
26
- "torch_dtype": "float32",
27
  "transformers_version": "4.37.0",
28
  "use_cache": true,
29
  "vocab_size": 51200
 
23
  "rope_scaling": null,
24
  "rope_theta": 10000.0,
25
  "tie_word_embeddings": false,
26
+ "torch_dtype": "float16",
27
  "transformers_version": "4.37.0",
28
  "use_cache": true,
29
  "vocab_size": 51200