gxy commited on
Commit
e4f529d
1 Parent(s): b3988c0

FEAT: first commit

Browse files
.gitattributes CHANGED
@@ -30,3 +30,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
30
  *.zip filter=lfs diff=lfs merge=lfs -text
31
  *.zst filter=lfs diff=lfs merge=lfs -text
32
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
30
  *.zip filter=lfs diff=lfs merge=lfs -text
31
  *.zst filter=lfs diff=lfs merge=lfs -text
32
  *tfevents* filter=lfs diff=lfs merge=lfs -text
33
+ pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -1,3 +1,49 @@
1
  ---
 
 
2
  license: apache-2.0
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - zh
4
  license: apache-2.0
5
+ tags:
6
+ - bart
7
+
8
+ widget:
9
+ - text: "桂林是著名的[MASK],它有很多[MASK]。"
10
+
11
+
12
  ---
13
+ # Randeng-BART-759M-BertTokenizer model (Chinese),one model of [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
14
+
15
+ The 759M million parameter Randeng-BART large model, using 180G Chinese data, 8 A100(40G) training for 7 days,which is a Encoder-Only transformer structure.
16
+
17
+ We use bert vocab as our tokenizer.
18
+
19
+ ## Task Description
20
+
21
+ Randeng-BART-759M-BertTokenizer is pre-trained by Text-Infilling task from BART [paper](https://readpaper.com/pdf-annotate/note?noteId=675945911766249472&pdfId=550970997159968917)
22
+
23
+ You can find our pretrain's code in [Fengshengbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM/tree/main/fengshen/examples/pretrain_randeng_bart)
24
+
25
+ ## Usage
26
+
27
+ ```python
28
+ from transformers import BartForConditionalGeneration, AutoTokenizer, Text2TextGenerationPipeline
29
+ import torch
30
+
31
+ tokenizer=AutoTokenizer.from_pretrained('IDEA-CCNL/Randeng-BART-759M-BertTokenizer', use_fast=false)
32
+ model=BartForConditionalGeneration.from_pretrained('IDEA-CCNL/Randeng-BART-759M-BertTokenizer')
33
+ text = '桂林是著名的[MASK],它有很多[MASK]。'
34
+ text2text_generator = Text2TextGenerationPipeline(model, tokenizer)
35
+ print(text2text_generator(text, max_length=50, do_sample=False))
36
+ ```
37
+
38
+ ## Citation
39
+
40
+ If you find the resource is useful, please cite the following website in your paper.
41
+
42
+ ```
43
+ @misc{Fengshenbang-LM,
44
+ title={Fengshenbang-LM},
45
+ author={IDEA-CCNL},
46
+ year={2022},
47
+ howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
48
+ }
49
+ ```
config.json ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bart-759M",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "BartForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 101,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 1024,
15
+ "d_kv": 64,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 24,
20
+ "decoder_start_token_id": 102,
21
+ "dropout": 0.1,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 24,
26
+ "eos_token_id": 102,
27
+ "forced_eos_token_id": 102,
28
+ "num_labels": 3,
29
+ "init_std": 0.02,
30
+ "is_encoder_decoder": true,
31
+ "max_position_embeddings": 1024,
32
+ "model_type": "bart",
33
+ "no_repeat_ngram_size": 3,
34
+ "normalize_before": false,
35
+ "normalize_embedding": true,
36
+ "num_beams": 4,
37
+ "num_hidden_layers": 24,
38
+ "pad_token_id": 0,
39
+ "scale_embedding": false,
40
+ "torch_dtype": "float16",
41
+ "transformers_version": "4.16.0.dev0",
42
+ "use_cache": true,
43
+ "vocab_size": 50265
44
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff970b3d11a9fa3c7a4d3f7a8694f0da3e56fb43605f73b1960a0c308d5237c4
3
+ size 1518240291
special_tokens_map.json ADDED
@@ -0,0 +1 @@
 
1
+ {"unk_token": "[UNK]", "sep_token": "[SEP]", "pad_token": "[PAD]", "cls_token": "[CLS]", "mask_token": "[MASK]"}
tokenizer_config.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "do_lower_case": true,
3
+ "do_basic_tokenize": true,
4
+ "never_split": null,
5
+ "unk_token": "[UNK]",
6
+ "sep_token": "[SEP]",
7
+ "pad_token": "[PAD]",
8
+ "cls_token": "[CLS]",
9
+ "mask_token": "[MASK]",
10
+ "tokenize_chinese_chars": true,
11
+ "strip_accents": null,
12
+ "special_tokens_map_file": null,
13
+ "name_or_path": "/cognitive_comp/gaoxinyu/pretrained_model/bert-1.3B",
14
+ "tokenizer_class": "BertTokenizer"
15
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff