groupuser/stablelm-2-1_6b
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

revert(config): use float16 default torch dtype

Browse Source
This commit is contained in:
Jonathan Tow 2024-02-21 20:41:28 +00:00 committed by system
parent ebf5f386bd
commit a2eb1af48d
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config.json
View File

@ -17,7 +17,7 @@
"partial_rotary_factor": 0.25,
"rope_theta": 10000,
"tie_word_embeddings": false,
"torch_dtype": "bfloat16",
"torch_dtype": "float16",
"transformers_version": "4.38.0",
"use_cache": true,
"use_qkv_bias": true,