Add pipeline tag and library name
Browse filesThis PR adds the `text-generation` pipeline tag and the `transformers` library name to the model card. This ensures the model can be found when searching by pipeline tag and makes the model page more informative.
README.md
CHANGED
@@ -1,12 +1,15 @@
|
|
1 |
---
|
2 |
-
|
|
|
3 |
datasets:
|
4 |
- hamishivi/gsm8k-symbolic
|
5 |
language:
|
6 |
- en
|
7 |
-
|
8 |
-
|
|
|
9 |
---
|
|
|
10 |
# TESS 2 v0.3 Symbolic - A Math-specific Tuned Diffusion LM
|
11 |
|
12 |
This model is the TESS 2 model trained on GSM8k symbolic data found [here](https://huggingface.co/datasets/hamishivi/gsm8k-symbolic), adapted from [here](https://github.com/HKUNLP/diffusion-of-thoughts). This model is a simplex-based diffusion model adapted from Mistral v0.1 7B, further trained on Dolma 1.7 and Tulu 2 SFT data.
|
|
|
1 |
---
|
2 |
+
base_model:
|
3 |
+
- hamishivi/tess2-v0.3-base
|
4 |
datasets:
|
5 |
- hamishivi/gsm8k-symbolic
|
6 |
language:
|
7 |
- en
|
8 |
+
license: apache-2.0
|
9 |
+
library_name: transformers
|
10 |
+
pipeline_tag: text-generation
|
11 |
---
|
12 |
+
|
13 |
# TESS 2 v0.3 Symbolic - A Math-specific Tuned Diffusion LM
|
14 |
|
15 |
This model is the TESS 2 model trained on GSM8k symbolic data found [here](https://huggingface.co/datasets/hamishivi/gsm8k-symbolic), adapted from [here](https://github.com/HKUNLP/diffusion-of-thoughts). This model is a simplex-based diffusion model adapted from Mistral v0.1 7B, further trained on Dolma 1.7 and Tulu 2 SFT data.
|