ssmits commited on
Commit
51a3096
1 Parent(s): 4bfaedb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +30 -28
README.md CHANGED
@@ -1,21 +1,32 @@
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
- # tiiuae/falcon-11B
13
 
14
- tiiuae/falcon-11B 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:
@@ -29,27 +40,18 @@ 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/tiiuae/falcon-11B"
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
- ```
 
1
  ---
2
+ base_model:
3
+ - tiiuae/falcon-11B
4
+ library_name: transformers
5
  tags:
 
6
  - mergekit
7
+ - merge
8
  - lazymergekit
9
+ license: apache-2.0
10
+ language:
11
+ - fr
 
12
  ---
13
+ # sliced
14
 
15
+ This is a merge of pre-trained language models created using [mergekit](https://github.com/cg123/mergekit).
16
 
17
+ ## Merge Details
18
+ ### Merge Method
19
+
20
+ This model was merged using the passthrough merge method.
21
+
22
+ ### Models Merged
23
+
24
+ The following models were included in the merge:
25
  * [tiiuae/falcon-11B](https://huggingface.co/tiiuae/falcon-11B)
26
 
27
+ ### Configuration
28
+
29
+ The following YAML configuration was used to produce this model:
30
 
31
  ```yaml
32
  slices:
 
40
  dtype: bfloat16
41
  ```
42
 
43
+ [PruneMe](https://github.com/arcee-ai/PruneMe) has been utilized using the wikimedia/wikipedia French (fr) subset by investigating layer similarity with 2000 samples. The layer ranges for pruning were determined based on this analysis to maintain performance while reducing model size.
44
 
45
+ ![Layer Similarity Plot](https://cdn-uploads.huggingface.co/production/uploads/660c0a02cf274b3ab77dd6b7/aS5jGo6KLv6BsmW_aO4PB.png)
 
46
 
47
+ ## Direct Use
48
+ Research on large language models; as a foundation for further specialization and finetuning for specific usecases (e.g., summarization, text generation, chatbot, etc.)
 
49
 
50
+ ## Out-of-Scope Use
51
+ Production use without adequate assessment of risks and mitigation; any use cases which may be considered irresponsible or harmful.
52
 
53
+ ## Bias, Risks, and Limitations
54
+ Falcon2-5.5B is trained mostly on English, but also German, Spanish, French, Italian, Portuguese, Polish, Dutch, Romanian, Czech, Swedish. It will not generalize appropriately to other languages. Furthermore, as it is trained on a large-scale corpora representative of the web, it will carry the stereotypes and biases commonly encountered online.
 
 
 
 
 
 
55
 
56
+ ## Recommendations
57
+ We recommend users of Falcon2-5.5B to consider finetuning it for the specific set of tasks of interest, and for guardrails and appropriate precautions to be taken for any production use.