ssmits commited on
Commit
7b18fc4
1 Parent(s): 7916731

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - merge
4
+ - mergekit
5
+ - lazymergekit
6
+ - tiiuae/falcon-11B
7
+ base_model:
8
+ - tiiuae/falcon-11B
9
+ - tiiuae/falcon-11B
10
+ ---
11
+
12
+ # Falcon2-5.5B-Swedish
13
+
14
+ Falcon2-5.5B-Swedish is a merge of the following models using [LazyMergekit](https://colab.research.google.com/drive/1obulZ1ROXHjYLn6PPZJwRR6GzgQogxxb?usp=sharing):
15
+ * [tiiuae/falcon-11B](https://huggingface.co/tiiuae/falcon-11B)
16
+ * [tiiuae/falcon-11B](https://huggingface.co/tiiuae/falcon-11B)
17
+
18
+ ## 🧩 Configuration
19
+
20
+ ```yaml
21
+ slices:
22
+ - sources:
23
+ - model: tiiuae/falcon-11B
24
+ layer_range: [0, 25]
25
+ - sources:
26
+ - model: tiiuae/falcon-11B
27
+ layer_range: [56, 59]
28
+ merge_method: passthrough
29
+ dtype: bfloat16
30
+ ```
31
+
32
+ ## 💻 Usage
33
+
34
+ ```python
35
+ !pip install -qU transformers accelerate
36
+
37
+ from transformers import AutoTokenizer
38
+ import transformers
39
+ import torch
40
+
41
+ model = "ssmits/Falcon2-5.5B-Swedish"
42
+ messages = [{"role": "user", "content": "What is a large language model?"}]
43
+
44
+ tokenizer = AutoTokenizer.from_pretrained(model)
45
+ prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
46
+ pipeline = transformers.pipeline(
47
+ "text-generation",
48
+ model=model,
49
+ torch_dtype=torch.float16,
50
+ device_map="auto",
51
+ )
52
+
53
+ outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
54
+ print(outputs[0]["generated_text"])
55
+ ```
config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "tiiuae/falcon-11B",
3
+ "activation": "gelu",
4
+ "alibi": false,
5
+ "architectures": [
6
+ "FalconForCausalLM"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "auto_map": {
10
+ "AutoConfig": "tiiuae/falcon-11B--configuration_falcon.FalconConfig",
11
+ "AutoModel": "tiiuae/falcon-11B--modeling_falcon.FalconModel",
12
+ "AutoModelForCausalLM": "tiiuae/falcon-11B--modeling_falcon.FalconForCausalLM",
13
+ "AutoModelForQuestionAnswering": "tiiuae/falcon-11B--modeling_falcon.FalconForQuestionAnswering",
14
+ "AutoModelForSequenceClassification": "tiiuae/falcon-11B--modeling_falcon.FalconForSequenceClassification",
15
+ "AutoModelForTokenClassification": "tiiuae/falcon-11B--modeling_falcon.FalconForTokenClassification"
16
+ },
17
+ "bias": false,
18
+ "bos_token_id": 11,
19
+ "eos_token_id": 11,
20
+ "ff_factor": 4,
21
+ "ffn_hidden_size": 16384,
22
+ "hidden_dropout": 0.0,
23
+ "hidden_size": 4096,
24
+ "initializer_range": 0.02,
25
+ "layer_norm_epsilon": 1e-05,
26
+ "max_position_embeddings": 8192,
27
+ "model_type": "falcon",
28
+ "multi_query": true,
29
+ "new_decoder_architecture": true,
30
+ "num_attention_heads": 32,
31
+ "num_hidden_layers": 28,
32
+ "num_kv_heads": 8,
33
+ "num_ln_in_parallel_attn": 1,
34
+ "parallel_attn": true,
35
+ "rope_scaling": null,
36
+ "rope_theta": 500042.0,
37
+ "tie_word_embeddings": false,
38
+ "torch_dtype": "bfloat16",
39
+ "transformers_version": "4.40.2",
40
+ "use_cache": true,
41
+ "vocab_size": 65024
42
+ }
mergekit_config.yml ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ slices:
3
+ - sources:
4
+ - model: tiiuae/falcon-11B
5
+ layer_range: [0, 25]
6
+ - sources:
7
+ - model: tiiuae/falcon-11B
8
+ layer_range: [56, 59]
9
+ merge_method: passthrough
10
+ dtype: bfloat16
model-00001-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:be473743446f6310af51e0c376f85362497f956b299ae82088be850871ebe058
3
+ size 885031960
model-00002-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7958aad384a953d794c245dc1fb049cf2e88db6555ebef8a37db39727a4f268
3
+ size 973112984
model-00003-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:742281f7d38f9ea479d6ad1458904c57428323cfe147476a9fc881340d3ea5a2
3
+ size 922798072
model-00004-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8bf29decce4e70a2158383597b1821eece07e06f1f58d629d62a30f47ce30bb2
3
+ size 922798080
model-00005-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1926ea5d345d3aefefe565f668d2c1e61e09bc21ed0a3e2b7ef371643a55deed
3
+ size 973112992
model-00006-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:612dc205bbe5d2e8b06a9dcd4c34aeb7bd041b59786defea49fe20c62ad08759
3
+ size 922798064
model-00007-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9b49e0a01cb4bf426c6781a0018642b2b0fb903ddfa45c2f1cc922015b4f0ebc
3
+ size 922798080
model-00008-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:46663363935d36efe732f818a0189aa38f01fd34c466b8162870cb8913c0d5e1
3
+ size 973112976
model-00009-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a997ff817469d17ed339b301fd69005a6cc071bddb9210e041cf868572fd22fd
3
+ size 922798056
model-00010-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fdc8f6860b3cc7296be0bdbc6e39218c54003d371ccb211394aa64d0fb3d7b3
3
+ size 922798072
model-00011-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:85e48d70ad97bd2153b3a7c46b69fe2e65fa58067a4946e66a267810d0044857
3
+ size 973112976
model-00012-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c287b5e845465167828d1374df8f4ccfa69a9d26eed0d02a26c7afb2b22e5e4a
3
+ size 616579648
model.safetensors.index.json ADDED
@@ -0,0 +1 @@
 
 
1
+ {"metadata": {"mergekit_version": "0.0.4.2", "total_size": 10930831360}, "weight_map": {"lm_head.weight": "model-00001-of-00012.safetensors", "transformer.h.0.input_layernorm.bias": "model-00001-of-00012.safetensors", "transformer.h.0.input_layernorm.weight": "model-00001-of-00012.safetensors", "transformer.h.0.mlp.dense_4h_to_h.weight": "model-00001-of-00012.safetensors", "transformer.h.0.mlp.dense_h_to_4h.weight": "model-00001-of-00012.safetensors", "transformer.h.0.self_attention.dense.weight": "model-00001-of-00012.safetensors", "transformer.h.0.self_attention.query_key_value.weight": "model-00001-of-00012.safetensors", "transformer.h.1.input_layernorm.bias": "model-00001-of-00012.safetensors", "transformer.h.1.input_layernorm.weight": "model-00001-of-00012.safetensors", "transformer.h.1.mlp.dense_4h_to_h.weight": "model-00002-of-00012.safetensors", "transformer.h.1.mlp.dense_h_to_4h.weight": "model-00002-of-00012.safetensors", "transformer.h.1.self_attention.dense.weight": "model-00002-of-00012.safetensors", "transformer.h.1.self_attention.query_key_value.weight": "model-00002-of-00012.safetensors", "transformer.h.10.input_layernorm.bias": "model-00002-of-00012.safetensors", "transformer.h.10.input_layernorm.weight": "model-00002-of-00012.safetensors", "transformer.h.10.mlp.dense_4h_to_h.weight": "model-00002-of-00012.safetensors", "transformer.h.10.mlp.dense_h_to_4h.weight": "model-00002-of-00012.safetensors", "transformer.h.10.self_attention.dense.weight": "model-00002-of-00012.safetensors", "transformer.h.10.self_attention.query_key_value.weight": "model-00002-of-00012.safetensors", "transformer.h.11.input_layernorm.bias": "model-00002-of-00012.safetensors", "transformer.h.11.input_layernorm.weight": "model-00002-of-00012.safetensors", "transformer.h.11.mlp.dense_4h_to_h.weight": "model-00002-of-00012.safetensors", "transformer.h.11.mlp.dense_h_to_4h.weight": "model-00002-of-00012.safetensors", "transformer.h.11.self_attention.dense.weight": "model-00003-of-00012.safetensors", "transformer.h.11.self_attention.query_key_value.weight": "model-00003-of-00012.safetensors", "transformer.h.12.input_layernorm.bias": "model-00003-of-00012.safetensors", "transformer.h.12.input_layernorm.weight": "model-00003-of-00012.safetensors", "transformer.h.12.mlp.dense_4h_to_h.weight": "model-00003-of-00012.safetensors", "transformer.h.12.mlp.dense_h_to_4h.weight": "model-00003-of-00012.safetensors", "transformer.h.12.self_attention.dense.weight": "model-00003-of-00012.safetensors", "transformer.h.12.self_attention.query_key_value.weight": "model-00003-of-00012.safetensors", "transformer.h.13.input_layernorm.bias": "model-00003-of-00012.safetensors", "transformer.h.13.input_layernorm.weight": "model-00003-of-00012.safetensors", "transformer.h.13.mlp.dense_4h_to_h.weight": "model-00003-of-00012.safetensors", "transformer.h.13.mlp.dense_h_to_4h.weight": "model-00003-of-00012.safetensors", "transformer.h.13.self_attention.dense.weight": "model-00003-of-00012.safetensors", "transformer.h.13.self_attention.query_key_value.weight": "model-00003-of-00012.safetensors", "transformer.h.14.input_layernorm.bias": "model-00003-of-00012.safetensors", "transformer.h.14.input_layernorm.weight": "model-00003-of-00012.safetensors", "transformer.h.14.mlp.dense_4h_to_h.weight": "model-00003-of-00012.safetensors", "transformer.h.14.mlp.dense_h_to_4h.weight": "model-00004-of-00012.safetensors", "transformer.h.14.self_attention.dense.weight": "model-00004-of-00012.safetensors", "transformer.h.14.self_attention.query_key_value.weight": "model-00004-of-00012.safetensors", "transformer.h.15.input_layernorm.bias": "model-00004-of-00012.safetensors", "transformer.h.15.input_layernorm.weight": "model-00004-of-00012.safetensors", "transformer.h.15.mlp.dense_4h_to_h.weight": "model-00004-of-00012.safetensors", "transformer.h.15.mlp.dense_h_to_4h.weight": "model-00004-of-00012.safetensors", "transformer.h.15.self_attention.dense.weight": "model-00004-of-00012.safetensors", "transformer.h.15.self_attention.query_key_value.weight": "model-00004-of-00012.safetensors", "transformer.h.16.input_layernorm.bias": "model-00004-of-00012.safetensors", "transformer.h.16.input_layernorm.weight": "model-00004-of-00012.safetensors", "transformer.h.16.mlp.dense_4h_to_h.weight": "model-00004-of-00012.safetensors", "transformer.h.16.mlp.dense_h_to_4h.weight": "model-00004-of-00012.safetensors", "transformer.h.16.self_attention.dense.weight": "model-00004-of-00012.safetensors", "transformer.h.16.self_attention.query_key_value.weight": "model-00004-of-00012.safetensors", "transformer.h.17.input_layernorm.bias": "model-00004-of-00012.safetensors", "transformer.h.17.input_layernorm.weight": "model-00004-of-00012.safetensors", "transformer.h.17.mlp.dense_4h_to_h.weight": "model-00005-of-00012.safetensors", "transformer.h.17.mlp.dense_h_to_4h.weight": "model-00005-of-00012.safetensors", "transformer.h.17.self_attention.dense.weight": "model-00005-of-00012.safetensors", "transformer.h.17.self_attention.query_key_value.weight": "model-00005-of-00012.safetensors", "transformer.h.18.input_layernorm.bias": "model-00005-of-00012.safetensors", "transformer.h.18.input_layernorm.weight": "model-00005-of-00012.safetensors", "transformer.h.18.mlp.dense_4h_to_h.weight": "model-00005-of-00012.safetensors", "transformer.h.18.mlp.dense_h_to_4h.weight": "model-00005-of-00012.safetensors", "transformer.h.18.self_attention.dense.weight": "model-00005-of-00012.safetensors", "transformer.h.18.self_attention.query_key_value.weight": "model-00005-of-00012.safetensors", "transformer.h.19.input_layernorm.bias": "model-00005-of-00012.safetensors", "transformer.h.19.input_layernorm.weight": "model-00005-of-00012.safetensors", "transformer.h.19.mlp.dense_4h_to_h.weight": "model-00005-of-00012.safetensors", "transformer.h.19.mlp.dense_h_to_4h.weight": "model-00005-of-00012.safetensors", "transformer.h.19.self_attention.dense.weight": "model-00006-of-00012.safetensors", "transformer.h.19.self_attention.query_key_value.weight": "model-00006-of-00012.safetensors", "transformer.h.2.input_layernorm.bias": "model-00006-of-00012.safetensors", "transformer.h.2.input_layernorm.weight": "model-00006-of-00012.safetensors", "transformer.h.2.mlp.dense_4h_to_h.weight": "model-00006-of-00012.safetensors", "transformer.h.2.mlp.dense_h_to_4h.weight": "model-00006-of-00012.safetensors", "transformer.h.2.self_attention.dense.weight": "model-00006-of-00012.safetensors", "transformer.h.2.self_attention.query_key_value.weight": "model-00006-of-00012.safetensors", "transformer.h.20.input_layernorm.bias": "model-00006-of-00012.safetensors", "transformer.h.20.input_layernorm.weight": "model-00006-of-00012.safetensors", "transformer.h.20.mlp.dense_4h_to_h.weight": "model-00006-of-00012.safetensors", "transformer.h.20.mlp.dense_h_to_4h.weight": "model-00006-of-00012.safetensors", "transformer.h.20.self_attention.dense.weight": "model-00006-of-00012.safetensors", "transformer.h.20.self_attention.query_key_value.weight": "model-00006-of-00012.safetensors", "transformer.h.21.input_layernorm.bias": "model-00006-of-00012.safetensors", "transformer.h.21.input_layernorm.weight": "model-00006-of-00012.safetensors", "transformer.h.21.mlp.dense_4h_to_h.weight": "model-00006-of-00012.safetensors", "transformer.h.21.mlp.dense_h_to_4h.weight": "model-00007-of-00012.safetensors", "transformer.h.21.self_attention.dense.weight": "model-00007-of-00012.safetensors", "transformer.h.21.self_attention.query_key_value.weight": "model-00007-of-00012.safetensors", "transformer.h.22.input_layernorm.bias": "model-00007-of-00012.safetensors", "transformer.h.22.input_layernorm.weight": "model-00007-of-00012.safetensors", "transformer.h.22.mlp.dense_4h_to_h.weight": "model-00007-of-00012.safetensors", "transformer.h.22.mlp.dense_h_to_4h.weight": "model-00007-of-00012.safetensors", "transformer.h.22.self_attention.dense.weight": "model-00007-of-00012.safetensors", "transformer.h.22.self_attention.query_key_value.weight": "model-00007-of-00012.safetensors", "transformer.h.23.input_layernorm.bias": "model-00007-of-00012.safetensors", "transformer.h.23.input_layernorm.weight": "model-00007-of-00012.safetensors", "transformer.h.23.mlp.dense_4h_to_h.weight": "model-00007-of-00012.safetensors", "transformer.h.23.mlp.dense_h_to_4h.weight": "model-00007-of-00012.safetensors", "transformer.h.23.self_attention.dense.weight": "model-00007-of-00012.safetensors", "transformer.h.23.self_attention.query_key_value.weight": "model-00007-of-00012.safetensors", "transformer.h.24.input_layernorm.bias": "model-00007-of-00012.safetensors", "transformer.h.24.input_layernorm.weight": "model-00007-of-00012.safetensors", "transformer.h.24.mlp.dense_4h_to_h.weight": "model-00008-of-00012.safetensors", "transformer.h.24.mlp.dense_h_to_4h.weight": "model-00008-of-00012.safetensors", "transformer.h.24.self_attention.dense.weight": "model-00008-of-00012.safetensors", "transformer.h.24.self_attention.query_key_value.weight": "model-00008-of-00012.safetensors", "transformer.h.3.input_layernorm.bias": "model-00008-of-00012.safetensors", "transformer.h.3.input_layernorm.weight": "model-00008-of-00012.safetensors", "transformer.h.3.mlp.dense_4h_to_h.weight": "model-00008-of-00012.safetensors", "transformer.h.3.mlp.dense_h_to_4h.weight": "model-00008-of-00012.safetensors", "transformer.h.3.self_attention.dense.weight": "model-00008-of-00012.safetensors", "transformer.h.3.self_attention.query_key_value.weight": "model-00008-of-00012.safetensors", "transformer.h.4.input_layernorm.bias": "model-00008-of-00012.safetensors", "transformer.h.4.input_layernorm.weight": "model-00008-of-00012.safetensors", "transformer.h.4.mlp.dense_4h_to_h.weight": "model-00008-of-00012.safetensors", "transformer.h.4.mlp.dense_h_to_4h.weight": "model-00008-of-00012.safetensors", "transformer.h.4.self_attention.dense.weight": "model-00009-of-00012.safetensors", "transformer.h.4.self_attention.query_key_value.weight": "model-00009-of-00012.safetensors", "transformer.h.5.input_layernorm.bias": "model-00009-of-00012.safetensors", "transformer.h.5.input_layernorm.weight": "model-00009-of-00012.safetensors", "transformer.h.5.mlp.dense_4h_to_h.weight": "model-00009-of-00012.safetensors", "transformer.h.5.mlp.dense_h_to_4h.weight": "model-00009-of-00012.safetensors", "transformer.h.5.self_attention.dense.weight": "model-00009-of-00012.safetensors", "transformer.h.5.self_attention.query_key_value.weight": "model-00009-of-00012.safetensors", "transformer.h.25.input_layernorm.bias": "model-00009-of-00012.safetensors", "transformer.h.25.input_layernorm.weight": "model-00009-of-00012.safetensors", "transformer.h.25.mlp.dense_4h_to_h.weight": "model-00009-of-00012.safetensors", "transformer.h.25.mlp.dense_h_to_4h.weight": "model-00009-of-00012.safetensors", "transformer.h.25.self_attention.dense.weight": "model-00009-of-00012.safetensors", "transformer.h.25.self_attention.query_key_value.weight": "model-00009-of-00012.safetensors", "transformer.h.26.input_layernorm.bias": "model-00009-of-00012.safetensors", "transformer.h.26.input_layernorm.weight": "model-00009-of-00012.safetensors", "transformer.h.26.mlp.dense_4h_to_h.weight": "model-00009-of-00012.safetensors", "transformer.h.26.mlp.dense_h_to_4h.weight": "model-00010-of-00012.safetensors", "transformer.h.26.self_attention.dense.weight": "model-00010-of-00012.safetensors", "transformer.h.26.self_attention.query_key_value.weight": "model-00010-of-00012.safetensors", "transformer.h.27.input_layernorm.bias": "model-00010-of-00012.safetensors", "transformer.h.27.input_layernorm.weight": "model-00010-of-00012.safetensors", "transformer.h.27.mlp.dense_4h_to_h.weight": "model-00010-of-00012.safetensors", "transformer.h.27.mlp.dense_h_to_4h.weight": "model-00010-of-00012.safetensors", "transformer.h.27.self_attention.dense.weight": "model-00010-of-00012.safetensors", "transformer.h.27.self_attention.query_key_value.weight": "model-00010-of-00012.safetensors", "transformer.h.6.input_layernorm.bias": "model-00010-of-00012.safetensors", "transformer.h.6.input_layernorm.weight": "model-00010-of-00012.safetensors", "transformer.h.6.mlp.dense_4h_to_h.weight": "model-00010-of-00012.safetensors", "transformer.h.6.mlp.dense_h_to_4h.weight": "model-00010-of-00012.safetensors", "transformer.h.6.self_attention.dense.weight": "model-00010-of-00012.safetensors", "transformer.h.6.self_attention.query_key_value.weight": "model-00010-of-00012.safetensors", "transformer.h.7.input_layernorm.bias": "model-00010-of-00012.safetensors", "transformer.h.7.input_layernorm.weight": "model-00010-of-00012.safetensors", "transformer.h.7.mlp.dense_4h_to_h.weight": "model-00011-of-00012.safetensors", "transformer.h.7.mlp.dense_h_to_4h.weight": "model-00011-of-00012.safetensors", "transformer.h.7.self_attention.dense.weight": "model-00011-of-00012.safetensors", "transformer.h.7.self_attention.query_key_value.weight": "model-00011-of-00012.safetensors", "transformer.h.8.input_layernorm.bias": "model-00011-of-00012.safetensors", "transformer.h.8.input_layernorm.weight": "model-00011-of-00012.safetensors", "transformer.h.8.mlp.dense_4h_to_h.weight": "model-00011-of-00012.safetensors", "transformer.h.8.mlp.dense_h_to_4h.weight": "model-00011-of-00012.safetensors", "transformer.h.8.self_attention.dense.weight": "model-00011-of-00012.safetensors", "transformer.h.8.self_attention.query_key_value.weight": "model-00011-of-00012.safetensors", "transformer.h.9.input_layernorm.bias": "model-00011-of-00012.safetensors", "transformer.h.9.input_layernorm.weight": "model-00011-of-00012.safetensors", "transformer.h.9.mlp.dense_4h_to_h.weight": "model-00011-of-00012.safetensors", "transformer.h.9.mlp.dense_h_to_4h.weight": "model-00011-of-00012.safetensors", "transformer.h.9.self_attention.dense.weight": "model-00012-of-00012.safetensors", "transformer.h.9.self_attention.query_key_value.weight": "model-00012-of-00012.safetensors", "transformer.ln_f.bias": "model-00012-of-00012.safetensors", "transformer.ln_f.weight": "model-00012-of-00012.safetensors", "transformer.word_embeddings.weight": "model-00012-of-00012.safetensors"}}
special_tokens_map.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ ">>TITLE<<",
4
+ ">>ABSTRACT<<",
5
+ ">>INTRODUCTION<<",
6
+ ">>SUMMARY<<",
7
+ ">>COMMENT<<",
8
+ ">>ANSWER<<",
9
+ ">>QUESTION<<",
10
+ ">>DOMAIN<<",
11
+ ">>PREFIX<<",
12
+ ">>SUFFIX<<",
13
+ ">>MIDDLE<<"
14
+ ],
15
+ "bos_token": {
16
+ "content": ">>",
17
+ "lstrip": false,
18
+ "normalized": false,
19
+ "rstrip": false,
20
+ "single_word": false
21
+ },
22
+ "eos_token": {
23
+ "content": "<|endoftext|>",
24
+ "lstrip": false,
25
+ "normalized": false,
26
+ "rstrip": false,
27
+ "single_word": false
28
+ },
29
+ "pad_token": {
30
+ "content": "<|endoftext|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false
35
+ }
36
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,135 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": ">>TITLE<<",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": ">>ABSTRACT<<",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": ">>INTRODUCTION<<",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": ">>SUMMARY<<",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "4": {
37
+ "content": ">>COMMENT<<",
38
+ "lstrip": false,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "5": {
45
+ "content": ">>ANSWER<<",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ },
52
+ "6": {
53
+ "content": ">>QUESTION<<",
54
+ "lstrip": false,
55
+ "normalized": false,
56
+ "rstrip": false,
57
+ "single_word": false,
58
+ "special": true
59
+ },
60
+ "7": {
61
+ "content": ">>DOMAIN<<",
62
+ "lstrip": false,
63
+ "normalized": false,
64
+ "rstrip": false,
65
+ "single_word": false,
66
+ "special": true
67
+ },
68
+ "8": {
69
+ "content": ">>PREFIX<<",
70
+ "lstrip": false,
71
+ "normalized": false,
72
+ "rstrip": false,
73
+ "single_word": false,
74
+ "special": true
75
+ },
76
+ "9": {
77
+ "content": ">>SUFFIX<<",
78
+ "lstrip": false,
79
+ "normalized": false,
80
+ "rstrip": false,
81
+ "single_word": false,
82
+ "special": true
83
+ },
84
+ "10": {
85
+ "content": ">>MIDDLE<<",
86
+ "lstrip": false,
87
+ "normalized": false,
88
+ "rstrip": false,
89
+ "single_word": false,
90
+ "special": true
91
+ },
92
+ "11": {
93
+ "content": "<|endoftext|>",
94
+ "lstrip": false,
95
+ "normalized": false,
96
+ "rstrip": false,
97
+ "single_word": false,
98
+ "special": true
99
+ },
100
+ "500": {
101
+ "content": ">>",
102
+ "lstrip": false,
103
+ "normalized": false,
104
+ "rstrip": false,
105
+ "single_word": false,
106
+ "special": true
107
+ }
108
+ },
109
+ "additional_special_tokens": [
110
+ ">>TITLE<<",
111
+ ">>ABSTRACT<<",
112
+ ">>INTRODUCTION<<",
113
+ ">>SUMMARY<<",
114
+ ">>COMMENT<<",
115
+ ">>ANSWER<<",
116
+ ">>QUESTION<<",
117
+ ">>DOMAIN<<",
118
+ ">>PREFIX<<",
119
+ ">>SUFFIX<<",
120
+ ">>MIDDLE<<"
121
+ ],
122
+ "bos_token": ">>",
123
+ "chat_template": "{% for message in messages %}\n{% if message['role'] == 'user' %}\n{{ 'User: \n' + message['content'] }}\n{% elif message['role'] == 'system' %}\n{{ 'System: ' + message['content'] }}\n{% elif message['role'] == 'assistant' %}\n{{ 'Falcon:\n' + message['content']}}\n{% endif %}\n{% if loop.last and add_generation_prompt %}\n{{ 'Falcon:' }}\n{% endif %}\n{% endfor %}",
124
+ "clean_up_tokenization_spaces": true,
125
+ "device_map": "cuda:2",
126
+ "eos_token": "<|endoftext|>",
127
+ "model_input_names": [
128
+ "input_ids",
129
+ "attention_mask"
130
+ ],
131
+ "model_max_length": 1000000000000000019884624838656,
132
+ "pad_token": "<|endoftext|>",
133
+ "padding_side": "left",
134
+ "tokenizer_class": "PreTrainedTokenizerFast"
135
+ }