burtenshaw HF staff commited on
Commit
d1edbb5
1 Parent(s): bd14734

Model save

Browse files
README.md CHANGED
@@ -1,61 +1,57 @@
1
  ---
2
- library_name: peft
3
- license: llama3.2
4
  base_model: meta-llama/Llama-3.2-1B
 
 
5
  tags:
 
6
  - trl
7
  - sft
8
- - generated_from_trainer
9
- datasets:
10
- - generator
11
- model-index:
12
- - name: code-llama-3-2-1b-commerce
13
- results: []
14
  ---
15
 
16
- <!-- This model card has been generated automatically according to the information the Trainer had access to. You
17
- should probably proofread and complete it, then remove this comment. -->
18
-
19
- # code-llama-3-2-1b-commerce
20
-
21
- This model is a fine-tuned version of [meta-llama/Llama-3.2-1B](https://huggingface.co/meta-llama/Llama-3.2-1B) on the generator dataset.
22
 
23
- ## Model description
 
24
 
25
- More information needed
26
 
27
- ## Intended uses & limitations
 
28
 
29
- More information needed
 
 
 
 
30
 
31
- ## Training and evaluation data
32
 
33
- More information needed
34
 
35
- ## Training procedure
36
 
37
- ### Training hyperparameters
38
 
39
- The following hyperparameters were used during training:
40
- - learning_rate: 0.0002
41
- - train_batch_size: 1
42
- - eval_batch_size: 8
43
- - seed: 42
44
- - gradient_accumulation_steps: 8
45
- - total_train_batch_size: 8
46
- - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
47
- - lr_scheduler_type: constant
48
- - lr_scheduler_warmup_ratio: 0.03
49
- - num_epochs: 3
50
 
51
- ### Training results
 
 
 
 
52
 
 
53
 
54
 
55
- ### Framework versions
56
 
57
- - PEFT 0.12.0
58
- - Transformers 4.44.2
59
- - Pytorch 2.4.0+cu121
60
- - Datasets 2.21.0
61
- - Tokenizers 0.19.1
 
 
 
 
 
 
 
 
1
  ---
 
 
2
  base_model: meta-llama/Llama-3.2-1B
3
+ library_name: transformers
4
+ model_name: code-llama-3-2-1b-commerce
5
  tags:
6
+ - generated_from_trainer
7
  - trl
8
  - sft
9
+ licence: license
 
 
 
 
 
10
  ---
11
 
12
+ # Model Card for code-llama-3-2-1b-commerce
 
 
 
 
 
13
 
14
+ This model is a fine-tuned version of [meta-llama/Llama-3.2-1B](https://huggingface.co/meta-llama/Llama-3.2-1B).
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="burtenshaw/code-llama-3-2-1b-commerce", 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
+ This model was trained with SFT.
33
 
34
+ ### Framework versions
 
 
 
 
 
 
 
 
 
 
35
 
36
+ - TRL: 0.12.1
37
+ - Transformers: 4.46.3
38
+ - Pytorch: 2.5.1
39
+ - Datasets: 3.1.0
40
+ - Tokenizers: 0.20.3
41
 
42
+ ## Citations
43
 
44
 
 
45
 
46
+ Cite TRL as:
47
+
48
+ ```bibtex
49
+ @misc{vonwerra2022trl,
50
+ title = {{TRL: Transformer Reinforcement Learning}},
51
+ 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},
52
+ year = 2020,
53
+ journal = {GitHub repository},
54
+ publisher = {GitHub},
55
+ howpublished = {\url{https://github.com/huggingface/trl}}
56
+ }
57
+ ```
adapter_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:618acbf37717e0afa9e706ce804e19568b4314c191a3ad68cf5f2ed129365528
3
  size 1772140424
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0399629676fda78ed182d394f7775c75f2ebe919a0f9a19113a94c2750a4404c
3
  size 1772140424
runs/Nov20_09-28-49_25c36ac8fe9b/events.out.tfevents.1732094957.25c36ac8fe9b.13977.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cbc7ab78b6ee06a5522948adb00d339d2f53fcef676219b36bad8e6650e193a5
3
- size 4184
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5a3356dd05f7c65d53964989fd0fd0de4f205d4867a0370d6913cc8e3222ef43
3
+ size 6543