From 6e5a8cd5d0472fd6e95506f416f506592e2f739f Mon Sep 17 00:00:00 2001 From: Ray Date: Sat, 28 Oct 2023 11:40:50 +0000 Subject: [PATCH] Upload config.json with huggingface_hub --- config.json | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/config.json b/config.json index c27c563..3a3671c 100644 --- a/config.json +++ b/config.json @@ -1,5 +1,5 @@ { - "_name_or_path": "/home/user/data/data/base_model/AtoM-7B", + "_name_or_path": "/mnt/data/zhangzheng/data/AtoM-7B/checkpoint-56000", "architectures": [ "LlamaForCausalLM" ], @@ -20,8 +20,8 @@ "rms_norm_eps": 1e-05, "rope_scaling": null, "tie_word_embeddings": false, - "torch_dtype": "bfloat16", - "transformers_version": "4.29.0", + "torch_dtype": "float16", + "transformers_version": "4.31.0", "use_cache": true, "vocab_size": 65000 }