Model save
Browse files
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)
|
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
|
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
|
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},
|