ingeol commited on
Commit
e8b5186
·
verified ·
1 Parent(s): 9a4ae85

Training in progress, epoch 0

Browse files
Files changed (3) hide show
  1. config.json +28 -0
  2. training_args.bin +3 -0
  3. upload.py +16 -0
config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "ingeol/kosaul_v0.2",
3
+ "architectures": [
4
+ "LlamaForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "bos_token_id": 128000,
9
+ "eos_token_id": 128001,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 4096,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 14336,
14
+ "max_position_embeddings": 8192,
15
+ "model_type": "llama",
16
+ "num_attention_heads": 32,
17
+ "num_hidden_layers": 32,
18
+ "num_key_value_heads": 8,
19
+ "pretraining_tp": 1,
20
+ "rms_norm_eps": 1e-05,
21
+ "rope_scaling": null,
22
+ "rope_theta": 500000.0,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "float32",
25
+ "transformers_version": "4.40.1",
26
+ "use_cache": false,
27
+ "vocab_size": 128256
28
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbad7a51936e9d60cc0f7618e20fe0a56e387481ec70f21cb5b60f23b6a7a931
3
+ size 6651
upload.py ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+
4
+ ckpt = 'checkpoint-70'
5
+ origin_model_id = 'beomi/Llama-3-Open-Ko-8B'
6
+ model = AutoModelForCausalLM.from_pretrained(ckpt)
7
+ tokenizer = AutoTokenizer.from_pretrained(origin_model_id)
8
+
9
+ tokenizer.pad_token = tokenizer.eos_token
10
+ model.resize_token_embeddings(len(tokenizer))
11
+
12
+ # state_dict = get_fp32_state_dict_from_zero_checkpoint(ckpt)
13
+ # model.load_state_dict(state_dict)
14
+
15
+ tokenizer.push_to_hub('ingeol/kosaul_v0.2')
16
+ model.push_to_hub('ingeol/kosaul_v0.2')