Add pipeline tag and improve model card (#1)
Browse files- Add pipeline tag and improve model card (8c2b606fc85469d117d76e290a65e5c29e76ca4d)
- Make the README consistent for the model over iterations (ad63cc0daf9fa12a3a3d48861445b5942a12a77c)
Co-authored-by: Niels Rogge <[email protected]>
README.md
CHANGED
|
@@ -1,19 +1,17 @@
|
|
| 1 |
---
|
| 2 |
-
library_name: transformers
|
| 3 |
-
license: mit
|
| 4 |
datasets:
|
| 5 |
- HuggingFaceH4/ultrafeedback_binarized
|
| 6 |
language:
|
| 7 |
- en
|
|
|
|
|
|
|
|
|
|
| 8 |
---
|
| 9 |
|
| 10 |
-
<!-- This is a model released from the preprint: *[Bootstrapping Language Models with DPO Implicit Rewards](https://arxiv.org/abs/2406.09760)*. Please refer to our [repository](https://github.com/sail-sg/dice) for more details. -->
|
| 11 |
-
|
| 12 |
# Llama-3-Base-8B-DICE-Iter1
|
| 13 |
|
| 14 |
This model was developed using [Bootstrapping Language Models with DPO Implicit Rewards](https://arxiv.org/abs/2406.09760) (DICE) at iteration 1, based on the [princeton-nlp/Llama-3-Base-8B-SFT-DPO](https://huggingface.co/princeton-nlp/Llama-3-Base-8B-SFT-DPO) architecture as the starting point.
|
| 15 |
|
| 16 |
-
<!-- We utilized the prompt sets extracted from [HuggingFaceH4/ultrafeedback_binarized](https://huggingface.co/datasets/HuggingFaceH4/ultrafeedback_binarized). -->
|
| 17 |
|
| 18 |
## Links to Other Models
|
| 19 |
- [Llama-3-Base-8B-DICE-Iter1](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter1)
|
|
@@ -34,6 +32,9 @@ This model was developed using [Bootstrapping Language Models with DPO Implicit
|
|
| 34 |
|[Llama-3-Base-8B-DICE-Iter1](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter1) |25.08 |25.77
|
| 35 |
|[Llama-3-Base-8B-DICE-Iter2](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter2) |**27.55** |**30.99**
|
| 36 |
|
|
|
|
|
|
|
|
|
|
| 37 |
## Citation
|
| 38 |
|
| 39 |
```bibtex
|
|
|
|
| 1 |
---
|
|
|
|
|
|
|
| 2 |
datasets:
|
| 3 |
- HuggingFaceH4/ultrafeedback_binarized
|
| 4 |
language:
|
| 5 |
- en
|
| 6 |
+
library_name: transformers
|
| 7 |
+
license: mit
|
| 8 |
+
pipeline_tag: text-generation
|
| 9 |
---
|
| 10 |
|
|
|
|
|
|
|
| 11 |
# Llama-3-Base-8B-DICE-Iter1
|
| 12 |
|
| 13 |
This model was developed using [Bootstrapping Language Models with DPO Implicit Rewards](https://arxiv.org/abs/2406.09760) (DICE) at iteration 1, based on the [princeton-nlp/Llama-3-Base-8B-SFT-DPO](https://huggingface.co/princeton-nlp/Llama-3-Base-8B-SFT-DPO) architecture as the starting point.
|
| 14 |
|
|
|
|
| 15 |
|
| 16 |
## Links to Other Models
|
| 17 |
- [Llama-3-Base-8B-DICE-Iter1](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter1)
|
|
|
|
| 32 |
|[Llama-3-Base-8B-DICE-Iter1](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter1) |25.08 |25.77
|
| 33 |
|[Llama-3-Base-8B-DICE-Iter2](https://huggingface.co/sail/Llama-3-Base-8B-DICE-Iter2) |**27.55** |**30.99**
|
| 34 |
|
| 35 |
+
## Code
|
| 36 |
+
https://github.com/sail-sg/dice
|
| 37 |
+
|
| 38 |
## Citation
|
| 39 |
|
| 40 |
```bibtex
|