DJseo commited on
Commit
be28be0
·
verified ·
1 Parent(s): 039550e

End of training

Browse files
README.md CHANGED
@@ -1,55 +1,58 @@
1
- ---
2
- library_name: peft
3
- license: llama3.2
4
- base_model: meta-llama/Llama-3.2-1B
5
- tags:
6
- - generated_from_trainer
7
- model-index:
8
- - name: Llama3_CAPL_Model
9
- results: []
10
- ---
11
 
12
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
- should probably proofread and complete it, then remove this comment. -->
14
 
15
- # Llama3_CAPL_Model
 
16
 
17
- This model is a fine-tuned version of [meta-llama/Llama-3.2-1B](https://huggingface.co/meta-llama/Llama-3.2-1B) on an unknown dataset.
18
 
19
- ## Model description
 
20
 
21
- More information needed
 
 
 
 
22
 
23
- ## Intended uses & limitations
24
-
25
- More information needed
26
-
27
- ## Training and evaluation data
28
 
29
- More information needed
30
 
31
- ## Training procedure
32
 
33
- ### Training hyperparameters
34
 
35
- The following hyperparameters were used during training:
36
- - learning_rate: 2e-05
37
- - train_batch_size: 4
38
- - eval_batch_size: 8
39
- - seed: 42
40
- - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
41
- - lr_scheduler_type: linear
42
- - num_epochs: 3
43
- - mixed_precision_training: Native AMP
44
 
45
- ### Training results
 
 
 
 
46
 
 
47
 
48
 
49
- ### Framework versions
50
 
51
- - PEFT 0.14.0
52
- - Transformers 4.49.0
53
- - Pytorch 2.5.1+cu121
54
- - Datasets 3.4.1
55
- - Tokenizers 0.21.1
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: meta-llama/Llama-3.1-8B-Instruct
3
+ library_name: transformers
4
+ model_name: Llama3_CAPL_Model
5
+ tags:
6
+ - generated_from_trainer
7
+ - trl
8
+ - sft
9
+ licence: license
10
+ ---
11
 
12
+ # Model Card for Llama3_CAPL_Model
 
13
 
14
+ This model is a fine-tuned version of [meta-llama/Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct).
15
+ It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
+ ## Quick start
18
 
19
+ ```python
20
+ from transformers import pipeline
21
 
22
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
23
+ generator = pipeline("text-generation", model="DJseo/Llama3_CAPL_Model", device="cuda")
24
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
25
+ print(output["generated_text"])
26
+ ```
27
 
28
+ ## Training procedure
 
 
 
 
29
 
30
+
31
 
 
32
 
33
+ This model was trained with SFT.
34
 
35
+ ### Framework versions
 
 
 
 
 
 
 
 
36
 
37
+ - TRL: 0.15.2
38
+ - Transformers: 4.49.0
39
+ - Pytorch: 2.5.1+cu121
40
+ - Datasets: 3.4.1
41
+ - Tokenizers: 0.21.1
42
 
43
+ ## Citations
44
 
45
 
 
46
 
47
+ Cite TRL as:
48
+
49
+ ```bibtex
50
+ @misc{vonwerra2022trl,
51
+ title = {{TRL: Transformer Reinforcement Learning}},
52
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou챕dec},
53
+ year = 2020,
54
+ journal = {GitHub repository},
55
+ publisher = {GitHub},
56
+ howpublished = {\url{https://github.com/huggingface/trl}}
57
+ }
58
+ ```
adapter_config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "alpha_pattern": {},
3
  "auto_mapping": null,
4
- "base_model_name_or_path": "meta-llama/Llama-3.2-1B",
5
  "bias": "none",
6
  "eva_config": null,
7
  "exclude_modules": null,
@@ -14,7 +14,7 @@
14
  "loftq_config": {},
15
  "lora_alpha": 32,
16
  "lora_bias": false,
17
- "lora_dropout": 0.1,
18
  "megatron_config": null,
19
  "megatron_core": "megatron.core",
20
  "modules_to_save": null,
@@ -24,6 +24,7 @@
24
  "revision": null,
25
  "target_modules": [
26
  "q_proj",
 
27
  "v_proj"
28
  ],
29
  "task_type": "CAUSAL_LM",
 
1
  {
2
  "alpha_pattern": {},
3
  "auto_mapping": null,
4
+ "base_model_name_or_path": "meta-llama/Llama-3.1-8B-Instruct",
5
  "bias": "none",
6
  "eva_config": null,
7
  "exclude_modules": null,
 
14
  "loftq_config": {},
15
  "lora_alpha": 32,
16
  "lora_bias": false,
17
+ "lora_dropout": 0.01,
18
  "megatron_config": null,
19
  "megatron_core": "megatron.core",
20
  "modules_to_save": null,
 
24
  "revision": null,
25
  "target_modules": [
26
  "q_proj",
27
+ "k_proj",
28
  "v_proj"
29
  ],
30
  "task_type": "CAUSAL_LM",
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0eacdde8c13b9c9bf354a850b8b92ae1c2bc2686978b612ecf034bacb26c48ef
3
- size 6824216
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee247f66339d47394dfacbf46e714f26ee52276e4ff7d52f206dc276e0a59288
3
+ size 37774528
runs/Mar26_16-35-39_DESKTOP-OBIMAP9/events.out.tfevents.1742974540.DESKTOP-OBIMAP9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0e4b22d433c51ef60d722fe5c741c7a4c7c3a8b9c8b93a0bfb1a9e2df1fbf81
3
+ size 9372
runs/Mar26_16-48-59_DESKTOP-OBIMAP9/events.out.tfevents.1742975340.DESKTOP-OBIMAP9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ae8eda4e02ef0c43776f0fa8dadef34c6bd429f5c1200aa4d6905f90ef74618
3
+ size 9372
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:429bbba745755918276024acd6af29e9a32c50cdcaec7523e06dd1fe9e111ed5
3
- size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e159ef1cb62b64dac53e6d836dc5a6d7aa3f25c1da910bb33a0298ea3700fc66
3
+ size 5624