knarayan commited on
Commit
7cb7899
·
verified ·
1 Parent(s): 5766a78

End of training

Browse files
README.md CHANGED
@@ -1,58 +1,57 @@
1
  ---
 
 
2
  base_model: PY007/TinyLlama-1.1B-Chat-v0.3
3
- library_name: transformers
4
- model_name: model_colors_lora
5
  tags:
6
  - generated_from_trainer
7
- - trl
8
- - sft
9
- licence: license
10
  ---
11
 
12
- # Model Card for model_colors_lora
 
13
 
14
- This model is a fine-tuned version of [PY007/TinyLlama-1.1B-Chat-v0.3](https://huggingface.co/PY007/TinyLlama-1.1B-Chat-v0.3).
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="knarayan/model_colors_lora", 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
- [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/keshi8086-palo-alto-networks/huggingface/runs/smacpsw8)
31
 
 
32
 
33
- This model was trained with SFT.
34
 
35
- ### Framework versions
36
 
37
- - TRL: 0.15.0
38
- - Transformers: 4.48.2
39
- - Pytorch: 2.5.1+cu124
40
- - Datasets: 3.2.0
41
- - Tokenizers: 0.21.0
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
- ```
 
1
  ---
2
+ library_name: peft
3
+ license: apache-2.0
4
  base_model: PY007/TinyLlama-1.1B-Chat-v0.3
 
 
5
  tags:
6
  - generated_from_trainer
7
+ model-index:
8
+ - name: model_colors_lora
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
+ # model_colors_lora
 
16
 
17
+ This model is a fine-tuned version of [PY007/TinyLlama-1.1B-Chat-v0.3](https://huggingface.co/PY007/TinyLlama-1.1B-Chat-v0.3) on the None 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: 0.0002
37
+ - train_batch_size: 8
38
+ - eval_batch_size: 8
39
+ - seed: 42
40
+ - gradient_accumulation_steps: 4
41
+ - total_train_batch_size: 32
42
+ - optimizer: Use OptimizerNames.PAGED_ADAMW with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
43
+ - lr_scheduler_type: cosine
44
+ - training_steps: 200
45
+ - mixed_precision_training: Native AMP
46
 
47
+ ### Training results
48
 
49
 
50
+
51
+ ### Framework versions
52
+
53
+ - PEFT 0.14.0
54
+ - Transformers 4.48.2
55
+ - Pytorch 2.5.1+cu124
56
+ - Datasets 3.2.0
57
+ - Tokenizers 0.21.0
 
 
 
 
runs/Feb13_20-03-50_664c4e247bf4/events.out.tfevents.1739477031.664c4e247bf4.1574.5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6ff5e184943e66f48ed51a3914a59b893f143fc95d08380f54be90997c0c864b
3
- size 9821
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b63520dbc57f4afd5cbe0411d544bece97bb702b5fbd32cf2105e59e74b8b3d
3
+ size 10175