shoubing35 commited on
Commit
f3dd2fe
·
verified ·
1 Parent(s): 3c36e95

Model save

Browse files
Files changed (1) hide show
  1. README.md +3 -4
README.md CHANGED
@@ -1,6 +1,5 @@
1
  ---
2
  base_model: meta-llama/Llama-3.2-1B-Instruct
3
- datasets: shoubing35/ones_digit_sft_dataset
4
  library_name: transformers
5
  model_name: llama-1B-sft
6
  tags:
@@ -12,7 +11,7 @@ licence: license
12
 
13
  # Model Card for llama-1B-sft
14
 
15
- This model is a fine-tuned version of [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct) on the [shoubing35/ones_digit_sft_dataset](https://huggingface.co/datasets/shoubing35/ones_digit_sft_dataset) dataset.
16
  It has been trained using [TRL](https://github.com/huggingface/trl).
17
 
18
  ## Quick start
@@ -35,7 +34,7 @@ This model was trained with SFT.
35
 
36
  ### Framework versions
37
 
38
- - TRL: 0.17.0.dev0
39
  - Transformers: 4.51.3
40
  - Pytorch: 2.6.0+cu124
41
  - Datasets: 3.6.0
@@ -50,7 +49,7 @@ Cite TRL as:
50
  ```bibtex
51
  @misc{vonwerra2022trl,
52
  title = {{TRL: Transformer Reinforcement Learning}},
53
- 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},
54
  year = 2020,
55
  journal = {GitHub repository},
56
  publisher = {GitHub},
 
1
  ---
2
  base_model: meta-llama/Llama-3.2-1B-Instruct
 
3
  library_name: transformers
4
  model_name: llama-1B-sft
5
  tags:
 
11
 
12
  # Model Card for llama-1B-sft
13
 
14
+ This model is a fine-tuned version of [meta-llama/Llama-3.2-1B-Instruct](https://huggingface.co/meta-llama/Llama-3.2-1B-Instruct).
15
  It has been trained using [TRL](https://github.com/huggingface/trl).
16
 
17
  ## Quick start
 
34
 
35
  ### Framework versions
36
 
37
+ - TRL: 0.17.0
38
  - Transformers: 4.51.3
39
  - Pytorch: 2.6.0+cu124
40
  - Datasets: 3.6.0
 
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{\'e}dec},
53
  year = 2020,
54
  journal = {GitHub repository},
55
  publisher = {GitHub},