NMCxyz commited on
Commit
f6cefef
·
verified ·
1 Parent(s): 3cde988

Upload folder

Browse files
Files changed (4) hide show
  1. config.json +40 -0
  2. pytorch_model.bin +3 -0
  3. speech_projector.bin +3 -0
  4. trainer_state.json +70 -0
config.json ADDED
@@ -0,0 +1,40 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "/data1/speech/anhnmt2/cuongnm/EOT/Qwen2.5-0.5B-Instruct",
3
+ "architectures": [
4
+ "Qwen2ForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "bos_token_id": 151643,
8
+ "eos_token_id": 151645,
9
+ "eot_loss_weight": 1.0,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 896,
12
+ "initializer_range": 0.02,
13
+ "intermediate_size": 4864,
14
+ "max_position_embeddings": 32768,
15
+ "max_window_layers": 21,
16
+ "model_type": "asr_speech_qwen2",
17
+ "num_attention_heads": 14,
18
+ "num_hidden_layers": 24,
19
+ "num_key_value_heads": 2,
20
+ "rms_norm_eps": 1e-06,
21
+ "rope_scaling": null,
22
+ "rope_theta": 1000000.0,
23
+ "sliding_window": null,
24
+ "speech_encoder": "/data1/speech/anhnmt2/Speech2Speech/LLaMA-Omni/models/speech_encoder/whisper-medium",
25
+ "speech_encoder_ds_rate": 5,
26
+ "speech_encoder_hidden_size": 1024,
27
+ "speech_encoder_type": "whisper",
28
+ "speech_normalize": false,
29
+ "speech_projector_lr": null,
30
+ "speech_projector_type": "linear",
31
+ "tie_word_embeddings": true,
32
+ "tokenizer_model_max_length": 1024,
33
+ "tokenizer_padding_side": "right",
34
+ "torch_dtype": "bfloat16",
35
+ "transformers_version": "4.45.0",
36
+ "tune_speech_projector": true,
37
+ "use_cache": true,
38
+ "use_sliding_window": false,
39
+ "vocab_size": 151665
40
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b4f56782af56c89934eb38a1be3267a13f7aae677ec340e87184859a7f5d132f
3
+ size 1898665066
speech_projector.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ae82fd314aa12c049d22992eb3d0adcfe39ac590803e1900b1834282c485b9a
3
+ size 24649624
trainer_state.json ADDED
@@ -0,0 +1,70 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 1000,
6
+ "global_step": 42,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.7142857142857143,
13
+ "grad_norm": 6.295878887176514,
14
+ "learning_rate": 0.0001845190085543795,
15
+ "loss": 4.9016,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 1.4285714285714286,
20
+ "grad_norm": 4.4954118728637695,
21
+ "learning_rate": 0.00012000256937760445,
22
+ "loss": 2.7019,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 2.142857142857143,
27
+ "grad_norm": 4.040550231933594,
28
+ "learning_rate": 4.3193525326884435e-05,
29
+ "loss": 2.1165,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 2.857142857142857,
34
+ "grad_norm": 3.2141640186309814,
35
+ "learning_rate": 1.2949737362087156e-06,
36
+ "loss": 1.5267,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 3.0,
41
+ "step": 42,
42
+ "total_flos": 266392691539968.0,
43
+ "train_loss": 2.760472910744803,
44
+ "train_runtime": 117.3664,
45
+ "train_samples_per_second": 1.431,
46
+ "train_steps_per_second": 0.358
47
+ }
48
+ ],
49
+ "logging_steps": 10,
50
+ "max_steps": 42,
51
+ "num_input_tokens_seen": 0,
52
+ "num_train_epochs": 3,
53
+ "save_steps": 1000,
54
+ "stateful_callbacks": {
55
+ "TrainerControl": {
56
+ "args": {
57
+ "should_epoch_stop": false,
58
+ "should_evaluate": false,
59
+ "should_log": false,
60
+ "should_save": true,
61
+ "should_training_stop": true
62
+ },
63
+ "attributes": {}
64
+ }
65
+ },
66
+ "total_flos": 266392691539968.0,
67
+ "train_batch_size": 4,
68
+ "trial_name": null,
69
+ "trial_params": null
70
+ }