初始化项目,由ModelHub XC社区提供模型

Model: togethercomputer/GPT-JT-6B-v1
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-06-08 08:06:16 +08:00
commit b07d11ab61
11 changed files with 50461 additions and 0 deletions

50
.gitattributes vendored Normal file
View File

@@ -0,0 +1,50 @@
*.7z filter=lfs diff=lfs merge=lfs -text
*.arrow filter=lfs diff=lfs merge=lfs -text
*.bz2 filter=lfs diff=lfs merge=lfs -text
*.ftz filter=lfs diff=lfs merge=lfs -text
*.gz filter=lfs diff=lfs merge=lfs -text
*.h5 filter=lfs diff=lfs merge=lfs -text
*.joblib filter=lfs diff=lfs merge=lfs -text
*.lfs.* filter=lfs diff=lfs merge=lfs -text
*.msgpack filter=lfs diff=lfs merge=lfs -text
*.onnx filter=lfs diff=lfs merge=lfs -text
*.ot filter=lfs diff=lfs merge=lfs -text
*.parquet filter=lfs diff=lfs merge=lfs -text
*.pb filter=lfs diff=lfs merge=lfs -text
*.pt filter=lfs diff=lfs merge=lfs -text
*.pth filter=lfs diff=lfs merge=lfs -text
*.rar filter=lfs diff=lfs merge=lfs -text
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
*.tar.* filter=lfs diff=lfs merge=lfs -text
*.tflite filter=lfs diff=lfs merge=lfs -text
*.tgz filter=lfs diff=lfs merge=lfs -text
*.xz filter=lfs diff=lfs merge=lfs -text
*.zip filter=lfs diff=lfs merge=lfs -text
*.zstandard filter=lfs diff=lfs merge=lfs -text
*.tfevents* filter=lfs diff=lfs merge=lfs -text
*.db* filter=lfs diff=lfs merge=lfs -text
*.ark* filter=lfs diff=lfs merge=lfs -text
**/*ckpt*data* filter=lfs diff=lfs merge=lfs -text
**/*ckpt*.meta filter=lfs diff=lfs merge=lfs -text
**/*ckpt*.index filter=lfs diff=lfs merge=lfs -text
*.safetensors filter=lfs diff=lfs merge=lfs -text
*.ckpt filter=lfs diff=lfs merge=lfs -text
*.gguf* filter=lfs diff=lfs merge=lfs -text
*.ggml filter=lfs diff=lfs merge=lfs -text
*.llamafile* filter=lfs diff=lfs merge=lfs -text
*.pt2 filter=lfs diff=lfs merge=lfs -text
*.mlmodel filter=lfs diff=lfs merge=lfs -text
*.npy filter=lfs diff=lfs merge=lfs -text
*.npz filter=lfs diff=lfs merge=lfs -text
*.pickle filter=lfs diff=lfs merge=lfs -text
*.pkl filter=lfs diff=lfs merge=lfs -text
*.tar filter=lfs diff=lfs merge=lfs -text
*.wasm filter=lfs diff=lfs merge=lfs -text
*.zst filter=lfs diff=lfs merge=lfs -text
*tfevents* filter=lfs diff=lfs merge=lfs -text
pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
tokenizer.json filter=lfs diff=lfs merge=lfs -text

158
README.md Normal file
View File

@@ -0,0 +1,158 @@
---
datasets:
- natural_instructions
- the_pile
- cot
- Muennighoff/P3
inference:
parameters:
max_new_tokens: 5
temperature: 1.0
top_k: 1
license: apache-2.0
language:
- en
pipeline_tag: text-generation
widget:
-
example_title: "Sentiment Analysis"
text: |-
The task is to label the post's emotion as sadness, joy, love, anger, fear, or surprise.
Input: I'm feeling quite sad and sorry for myself but ill snap out of it soon.
Output: sadness
Input: I am just feeling cranky and blue.
Output: anger
Input: I can have for a treat or if i am feeling festive.
Output:
-
example_title: "Country Currency"
text: |-
Return the currency of the given country.
Input: Switzerland
Output: Swiss Franc
Input: India
Output:
-
example_title: "Tweet Eval Hate"
text: |-
Label whether the following tweet contains hate speech against either immigrants or women. Hate Speech (HS) is commonly defined as any communication that disparages a person or a group on the basis of some characteristic such as race, color, ethnicity, gender, sexual orientation, nationality, religion, or other characteristics.
Possible labels:
1. hate speech
2. not hate speech
Tweet: HOW REFRESHING! In South Korea, there is no such thing as 'political correctness" when it comes to dealing with Muslim refugee wannabes via @user
Label: hate speech
Tweet: New to Twitter-- any men on here know what the process is to get #verified?
Label: not hate speech
Tweet: Dont worry @user you are and will always be the most hysterical woman.
Label:
-
example_title: "Entity Recognition"
text: |-
Extract all the names of people, places, and organizations from the following sentences.
Sentence: Satya Nadella, the CEO of Microsoft, was visiting the Bahamas last May.
Entities: Satya Nadella, Microsoft, Bahamas
Sentence: Pacific Northwest cities include Seattle and Portland, which I have visited with Vikash.
Entities:
-
example_title: "Data Clearning"
text: |-
Format the data into a CSV file:
Input: Jane Doe jane.doe@gmail.com (520) 382 2435
Output: Jane Doe,jane.doe@gmail.com,520-382-2435
Input: Peter Lee (510) 333-2429 email: peter@yahoo.com
Output:
---
<h1 style="font-size: 42px">GPT-JT<h1/>
***<p style="font-size: 24px">Feel free to try out our [Online Demo](https://huggingface.co/spaces/togethercomputer/GPT-JT)!</p>***
# Model Summary
> With a new decentralized training algorithm, we fine-tuned GPT-J (6B) on 3.53 billion tokens, resulting in GPT-JT (6B), a model that outperforms many 100B+ parameter models on classification benchmarks.
We incorporated a collection of open techniques and datasets to build GPT-JT:
- GPT-JT is a fork of [EleutherAI](https://www.eleuther.ai)'s [GPT-J (6B)](https://huggingface.co/EleutherAI/gpt-j-6B);
- We used [UL2](https://github.com/google-research/google-research/tree/master/ul2)'s training objective, allowing the model to see bidirectional context of the prompt;
- The model was trained on a large collection of diverse data, including [Chain-of-Thought (CoT)](https://ai.googleblog.com/2022/05/language-models-perform-reasoning-via.html), [Public Pool of Prompts (P3) dataset](https://huggingface.co/datasets/bigscience/P3), [Natural-Instructions (NI) dataset](https://github.com/allenai/natural-instructions).
With the help of techniques mentioned above, GPT-JT significantly improves the performance of classification tasks over the original GPT-J, and even outperforms most 100B+ parameter models!
# Quick Start
```python
from transformers import pipeline
pipe = pipeline(model='togethercomputer/GPT-JT-6B-v1')
pipe('''"I love this!" Is it positive? A:''')
```
or
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("togethercomputer/GPT-JT-6B-v1")
model = AutoModelForCausalLM.from_pretrained("togethercomputer/GPT-JT-6B-v1")
```
# License
The weights of GPT-JT-6B-v1 are licensed under version 2.0 of the Apache License.
# Training Details
## UL2 Training Objective
We train GPT-JT using UL2 training objective [1][2].
The original GPT-J uses causal mask (as shown below left) for autoregressive generation. So for each token, it can only see its previous context.
In order to fully leverage the context information, we continue to train GPT-J with UL2 training objectives, and uses causal mask with prefix (as shown below right) -- using bidirectional attention for the prompt / input and causal attention for token generation.
Intuitively, being able to see context bidirectionally might improve downstream tasks that require this information.
$$
\begin{bmatrix}
1 & 0 & 0 & 0 & 0 \\
1 & 1 & 0 & 0 & 0 \\
1 & 1 & 1 & 0 & 0 \\
1 & 1 & 1 & 1 & 0 \\
1 & 1 & 1 & 1 & 1
\end{bmatrix}
\begin{bmatrix}
1 & 1 & 1 & 0 & 0 \\
1 & 1 & 1 & 0 & 0 \\
1 & 1 & 1 & 0 & 0 \\
1 & 1 & 1 & 1 & 0 \\
1 & 1 & 1 & 1 & 1
\end{bmatrix}
$$
Furthermore, we leverage a large collection of data, including [Natural-Instructions](https://github.com/allenai/natural-instructions), [P3](https://huggingface.co/datasets/Muennighoff/P3), [MMLU-COT](https://github.com/jasonwei20/flan-2/blob/main/mmlu-cot.json), and [the Pile](https://huggingface.co/datasets/the_pile)
Specifically, we first conduct training for 2.62 billion tokens using the UL2 loss on the Pile, followed by 0.92 billion tokens with a mixture of the above datasets: 5% of COT, 20% of P3, 20% of NI, and 55% of the Pile.
## Hyperparameters
We used AdamW with a learning rate of 1e-5 and global batch size of 64 (16 for each data parallel worker).
We used mix-precision training where the activation is in FP16 while the optimizer states are kept in FP32.
We use both data parallelism and pipeline parallelism to conduct training.
During training, we truncate the input sequence to 2048 tokens, and for input sequence that contains less than 2048 tokens, we concatenate multiple sequences into one long sequence to improve the data efficiency.
## Infrastructure
We used [the Together Research Computer](https://together.xyz/) to conduct training.
# References
[1]: Tay, Yi, Mostafa Dehghani, Vinh Q. Tran, Xavier Garcia, Dara Bahri, Tal Schuster, Huaixiu Steven Zheng, Neil Houlsby, and Donald Metzler. "Unifying Language Learning Paradigms." arXiv preprint arXiv:2205.05131 (2022).
[2]: Tay, Yi, Jason Wei, Hyung Won Chung, Vinh Q. Tran, David R. So, Siamak Shakeri, Xavier Garcia et al. "Transcending scaling laws with 0.1% extra compute." arXiv preprint arXiv:2210.11399 (2022).

145
added_tokens.json Normal file
View File

@@ -0,0 +1,145 @@
{
"<|extratoken_100|>": 50356,
"<|extratoken_101|>": 50357,
"<|extratoken_102|>": 50358,
"<|extratoken_103|>": 50359,
"<|extratoken_104|>": 50360,
"<|extratoken_105|>": 50361,
"<|extratoken_106|>": 50362,
"<|extratoken_107|>": 50363,
"<|extratoken_108|>": 50364,
"<|extratoken_109|>": 50365,
"<|extratoken_10|>": 50266,
"<|extratoken_110|>": 50366,
"<|extratoken_111|>": 50367,
"<|extratoken_112|>": 50368,
"<|extratoken_113|>": 50369,
"<|extratoken_114|>": 50370,
"<|extratoken_115|>": 50371,
"<|extratoken_116|>": 50372,
"<|extratoken_117|>": 50373,
"<|extratoken_118|>": 50374,
"<|extratoken_119|>": 50375,
"<|extratoken_11|>": 50267,
"<|extratoken_120|>": 50376,
"<|extratoken_121|>": 50377,
"<|extratoken_122|>": 50378,
"<|extratoken_123|>": 50379,
"<|extratoken_124|>": 50380,
"<|extratoken_125|>": 50381,
"<|extratoken_126|>": 50382,
"<|extratoken_127|>": 50383,
"<|extratoken_128|>": 50384,
"<|extratoken_129|>": 50385,
"<|extratoken_12|>": 50268,
"<|extratoken_130|>": 50386,
"<|extratoken_131|>": 50387,
"<|extratoken_132|>": 50388,
"<|extratoken_133|>": 50389,
"<|extratoken_134|>": 50390,
"<|extratoken_135|>": 50391,
"<|extratoken_136|>": 50392,
"<|extratoken_137|>": 50393,
"<|extratoken_138|>": 50394,
"<|extratoken_139|>": 50395,
"<|extratoken_13|>": 50269,
"<|extratoken_140|>": 50396,
"<|extratoken_141|>": 50397,
"<|extratoken_142|>": 50398,
"<|extratoken_143|>": 50399,
"<|extratoken_14|>": 50270,
"<|extratoken_15|>": 50271,
"<|extratoken_16|>": 50272,
"<|extratoken_17|>": 50273,
"<|extratoken_18|>": 50274,
"<|extratoken_19|>": 50275,
"<|extratoken_1|>": 50257,
"<|extratoken_20|>": 50276,
"<|extratoken_21|>": 50277,
"<|extratoken_22|>": 50278,
"<|extratoken_23|>": 50279,
"<|extratoken_24|>": 50280,
"<|extratoken_25|>": 50281,
"<|extratoken_26|>": 50282,
"<|extratoken_27|>": 50283,
"<|extratoken_28|>": 50284,
"<|extratoken_29|>": 50285,
"<|extratoken_2|>": 50258,
"<|extratoken_30|>": 50286,
"<|extratoken_31|>": 50287,
"<|extratoken_32|>": 50288,
"<|extratoken_33|>": 50289,
"<|extratoken_34|>": 50290,
"<|extratoken_35|>": 50291,
"<|extratoken_36|>": 50292,
"<|extratoken_37|>": 50293,
"<|extratoken_38|>": 50294,
"<|extratoken_39|>": 50295,
"<|extratoken_3|>": 50259,
"<|extratoken_40|>": 50296,
"<|extratoken_41|>": 50297,
"<|extratoken_42|>": 50298,
"<|extratoken_43|>": 50299,
"<|extratoken_44|>": 50300,
"<|extratoken_45|>": 50301,
"<|extratoken_46|>": 50302,
"<|extratoken_47|>": 50303,
"<|extratoken_48|>": 50304,
"<|extratoken_49|>": 50305,
"<|extratoken_4|>": 50260,
"<|extratoken_50|>": 50306,
"<|extratoken_51|>": 50307,
"<|extratoken_52|>": 50308,
"<|extratoken_53|>": 50309,
"<|extratoken_54|>": 50310,
"<|extratoken_55|>": 50311,
"<|extratoken_56|>": 50312,
"<|extratoken_57|>": 50313,
"<|extratoken_58|>": 50314,
"<|extratoken_59|>": 50315,
"<|extratoken_5|>": 50261,
"<|extratoken_60|>": 50316,
"<|extratoken_61|>": 50317,
"<|extratoken_62|>": 50318,
"<|extratoken_63|>": 50319,
"<|extratoken_64|>": 50320,
"<|extratoken_65|>": 50321,
"<|extratoken_66|>": 50322,
"<|extratoken_67|>": 50323,
"<|extratoken_68|>": 50324,
"<|extratoken_69|>": 50325,
"<|extratoken_6|>": 50262,
"<|extratoken_70|>": 50326,
"<|extratoken_71|>": 50327,
"<|extratoken_72|>": 50328,
"<|extratoken_73|>": 50329,
"<|extratoken_74|>": 50330,
"<|extratoken_75|>": 50331,
"<|extratoken_76|>": 50332,
"<|extratoken_77|>": 50333,
"<|extratoken_78|>": 50334,
"<|extratoken_79|>": 50335,
"<|extratoken_7|>": 50263,
"<|extratoken_80|>": 50336,
"<|extratoken_81|>": 50337,
"<|extratoken_82|>": 50338,
"<|extratoken_83|>": 50339,
"<|extratoken_84|>": 50340,
"<|extratoken_85|>": 50341,
"<|extratoken_86|>": 50342,
"<|extratoken_87|>": 50343,
"<|extratoken_88|>": 50344,
"<|extratoken_89|>": 50345,
"<|extratoken_8|>": 50264,
"<|extratoken_90|>": 50346,
"<|extratoken_91|>": 50347,
"<|extratoken_92|>": 50348,
"<|extratoken_93|>": 50349,
"<|extratoken_94|>": 50350,
"<|extratoken_95|>": 50351,
"<|extratoken_96|>": 50352,
"<|extratoken_97|>": 50353,
"<|extratoken_98|>": 50354,
"<|extratoken_99|>": 50355,
"<|extratoken_9|>": 50265
}

42
config.json Normal file
View File

@@ -0,0 +1,42 @@
{
"_name_or_path": "togethercomputer/GPT-JT-6B-v1",
"activation_function": "gelu_new",
"architectures": [
"GPTJForCausalLM"
],
"attn_pdrop": 0.0,
"bos_token_id": 50256,
"embd_pdrop": 0.0,
"eos_token_id": 50256,
"gradient_checkpointing": false,
"initializer_range": 0.02,
"layer_norm_epsilon": 1e-05,
"model_type": "gptj",
"n_embd": 4096,
"n_head": 16,
"n_inner": null,
"n_layer": 28,
"n_positions": 2048,
"resid_pdrop": 0.0,
"rotary": true,
"rotary_dim": 64,
"scale_attn_weights": true,
"summary_activation": null,
"summary_first_dropout": 0.1,
"summary_proj_to_labels": true,
"summary_type": "cls_index",
"summary_use_proj": true,
"task_specific_params": {
"text-generation": {
"do_sample": true,
"max_length": 50,
"temperature": 0.6
}
},
"tie_word_embeddings": false,
"tokenizer_class": "GPT2Tokenizer",
"torch_dtype": "float16",
"transformers_version": "4.21.1",
"use_cache": true,
"vocab_size": 50400
}

1
configuration.json Normal file
View File

@@ -0,0 +1 @@
{"framework": "pytorch", "task": "text-generation", "allow_remote": true}

50001
merges.txt Normal file

File diff suppressed because it is too large Load Diff

3
pytorch_model.bin Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:229dde9fafdbca6c993c719eba8b1e9fd9423edd482a4afdbd4d0a139f447b81
size 12219313951

23
special_tokens_map.json Normal file
View File

@@ -0,0 +1,23 @@
{
"bos_token": {
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
},
"eos_token": {
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
},
"unk_token": {
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
}
}

3
tokenizer.json Normal file
View File

@@ -0,0 +1,3 @@
version https://git-lfs.github.com/spec/v1
oid sha256:865b2d46c6da3702eb337c8b6636be40b059c5b3ab175e565800d2c8e7e09ebf
size 2135258

34
tokenizer_config.json Normal file
View File

@@ -0,0 +1,34 @@
{
"add_bos_token": false,
"add_prefix_space": false,
"bos_token": {
"__type": "AddedToken",
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
},
"eos_token": {
"__type": "AddedToken",
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
},
"errors": "replace",
"model_max_length": 2048,
"name_or_path": "EleutherAI/gpt-j-6B",
"pad_token": null,
"special_tokens_map_file": null,
"tokenizer_class": "GPT2Tokenizer",
"unk_token": {
"__type": "AddedToken",
"content": "<|endoftext|>",
"lstrip": false,
"normalized": true,
"rstrip": false,
"single_word": false
}
}

1
vocab.json Normal file

File diff suppressed because one or more lines are too long