updatedd readme
#1
by
devboby - opened
README.md
CHANGED
|
@@ -4,12 +4,10 @@ language:
|
|
| 4 |
- en
|
| 5 |
pipeline_tag: text-generation
|
| 6 |
tags:
|
| 7 |
-
- facebook
|
| 8 |
-
- meta
|
| 9 |
- pytorch
|
| 10 |
-
-
|
| 11 |
-
-
|
| 12 |
-
model_type:
|
| 13 |
license: other
|
| 14 |
---
|
| 15 |
|
|
@@ -17,17 +15,17 @@ license: other
|
|
| 17 |
|
| 18 |
<!-- header end -->
|
| 19 |
|
| 20 |
-
#
|
| 21 |
|
| 22 |
-
These files are GGML format model files for [
|
| 23 |
|
| 24 |
-
GGML files are for CPU + GPU inference using [
|
| 25 |
* [KoboldCpp](https://github.com/LostRuins/koboldcpp), a powerful GGML web UI with full GPU acceleration out of the box. Especially good for story telling.
|
| 26 |
* [LoLLMS Web UI](https://github.com/ParisNeo/lollms-webui), a great web UI with GPU acceleration via the c_transformers backend.
|
| 27 |
* [LM Studio](https://lmstudio.ai/), a fully featured local GUI. Supports full GPU accel on macOS. Also supports Windows, without GPU accel.
|
| 28 |
-
* [text-generation-webui](https://github.com/oobabooga/text-generation-webui), the most popular web UI. Requires extra steps to enable GPU accel via
|
| 29 |
* [ctransformers](https://github.com/marella/ctransformers), a Python library with LangChain support and OpenAI-compatible AI server.
|
| 30 |
-
* [
|
| 31 |
*
|
| 32 |
|
| 33 |
## Prompt template: None
|
|
@@ -39,15 +37,15 @@ GGML files are for CPU + GPU inference using [llama.cpp](https://github.com/gger
|
|
| 39 |
<!-- compatibility_ggml start -->
|
| 40 |
## Compatibility
|
| 41 |
|
| 42 |
-
### Original
|
| 43 |
|
| 44 |
These are guaranteed to be compatible with any UIs, tools and libraries released since late May. They may be phased out soon, as they are largely superseded by the new k-quant methods.
|
| 45 |
|
| 46 |
### New k-quant methods: `q2_K, q3_K_S, q3_K_M, q3_K_L, q4_K_S, q4_K_M, q5_K_S, q6_K`
|
| 47 |
|
| 48 |
-
These new quantisation methods are compatible with
|
| 49 |
|
| 50 |
-
They are now also compatible with recent releases of text-generation-webui, KoboldCpp,
|
| 51 |
|
| 52 |
## Explanation of the new k-quant methods
|
| 53 |
<details>
|
|
@@ -68,29 +66,29 @@ Refer to the Provided Files table below to see what files use which methods, and
|
|
| 68 |
## Provided files
|
| 69 |
| Name | Quant method | Bits | Size | Max RAM required | Use case |
|
| 70 |
| ---- | ---- | ---- | ---- | ---- | ----- |
|
| 71 |
-
|
|
| 72 |
-
|
|
| 73 |
-
|
|
| 74 |
-
|
|
| 75 |
-
|
|
| 76 |
-
|
|
| 77 |
-
|
|
| 78 |
-
|
|
| 79 |
-
|
|
| 80 |
-
|
|
| 81 |
-
|
|
| 82 |
-
|
|
| 83 |
-
|
|
| 84 |
-
|
|
| 85 |
|
| 86 |
**Note**: the above RAM figures assume no GPU offloading. If layers are offloaded to the GPU, this will reduce RAM usage and use VRAM instead.
|
| 87 |
|
| 88 |
-
## How to run in `
|
| 89 |
|
| 90 |
I use the following command line; adjust for your tastes and needs:
|
| 91 |
|
| 92 |
```
|
| 93 |
-
./main -t 10 -ngl 32 -m
|
| 94 |
```
|
| 95 |
Change `-t 10` to the number of physical CPU cores you have. For example if your system has 8 cores/16 threads, use `-t 8`.
|
| 96 |
|
|
@@ -100,142 +98,142 @@ If you want to have a chat-style conversation, replace the `-p <PROMPT>` argumen
|
|
| 100 |
|
| 101 |
## How to run in `text-generation-webui`
|
| 102 |
|
| 103 |
-
Further instructions here: [text-generation-webui/docs/
|
| 104 |
|
| 105 |
|
| 106 |
-
# Original model card:
|
| 107 |
|
| 108 |
---
|
| 109 |
-
extra_gated_heading: Access
|
| 110 |
extra_gated_description: >-
|
| 111 |
-
This is a form to enable access to
|
| 112 |
-
granted access from
|
| 113 |
license terms and acceptable use policy before submitting this form. Requests
|
| 114 |
will be processed in 1-2 days.
|
| 115 |
extra_gated_button_content: Submit
|
| 116 |
extra_gated_fields:
|
| 117 |
-
I agree to share my name, email address and username with
|
| 118 |
language:
|
| 119 |
- en
|
| 120 |
pipeline_tag: text-generation
|
| 121 |
inference: false
|
| 122 |
tags:
|
| 123 |
- facebook
|
| 124 |
-
-
|
| 125 |
- pytorch
|
| 126 |
-
-
|
| 127 |
-
-
|
| 128 |
---
|
| 129 |
-
# **
|
| 130 |
-
|
| 131 |
|
| 132 |
## Model Details
|
| 133 |
-
*Note: Use of this model is governed by the
|
| 134 |
|
| 135 |
-
|
| 136 |
|
| 137 |
-
**Model Developers**
|
| 138 |
|
| 139 |
-
**Variations**
|
| 140 |
|
| 141 |
**Input** Models input text only.
|
| 142 |
|
| 143 |
**Output** Models generate text only.
|
| 144 |
|
| 145 |
-
**Model Architecture**
|
| 146 |
|
| 147 |
|
| 148 |
||Training Data|Params|Content Length|GQA|Tokens|LR|
|
| 149 |
|---|---|---|---|---|---|---|
|
| 150 |
-
|
|
| 151 |
-
|
|
| 152 |
-
|
|
| 153 |
|
| 154 |
-
*
|
| 155 |
|
| 156 |
-
**Model Dates**
|
| 157 |
|
| 158 |
**Status** This is a static model trained on an offline dataset. Future versions of the tuned models will be released as we improve model safety with community feedback.
|
| 159 |
|
| 160 |
-
**License** A custom commercial license is available at: [https://ai
|
| 161 |
|
| 162 |
## Intended Use
|
| 163 |
-
**Intended Use Cases**
|
| 164 |
|
| 165 |
-
To get the expected features and performance for the chat versions, a specific formatting needs to be followed, including the `INST` and `<<SYS>>` tags, `BOS` and `EOS` tokens, and the whitespaces and breaklines in between (we recommend calling `strip()` on inputs to avoid double-spaces). See our reference code in github for details: [`chat_completion`](https://github.com/facebookresearch/
|
| 166 |
|
| 167 |
-
**Out-of-scope Uses** Use in any manner that violates applicable laws or regulations (including trade compliance laws).Use in languages other than English. Use in any other way that is prohibited by the Acceptable Use Policy and Licensing Agreement for
|
| 168 |
|
| 169 |
## Hardware and Software
|
| 170 |
-
**Training Factors** We used custom training libraries,
|
| 171 |
|
| 172 |
-
**Carbon Footprint** Pretraining utilized a cumulative 3.3M GPU hours of computation on hardware of type A100-80GB (TDP of 350-400W). Estimated total emissions were 539 tCO2eq, 100% of which were offset by
|
| 173 |
|
| 174 |
||Time (GPU hours)|Power Consumption (W)|Carbon Emitted(tCO<sub>2</sub>eq)|
|
| 175 |
|---|---|---|---|
|
| 176 |
-
|
|
| 177 |
-
|
|
| 178 |
-
|
|
| 179 |
|Total|3311616||539.00|
|
| 180 |
|
| 181 |
-
**CO<sub>2</sub> emissions during pretraining.** Time: total GPU time required for training each model. Power Consumption: peak power capacity per GPU device for the GPUs used adjusted for power usage efficiency. 100% of the emissions are directly offset by
|
| 182 |
|
| 183 |
## Training Data
|
| 184 |
-
**Overview**
|
| 185 |
|
| 186 |
**Data Freshness** The pretraining data has a cutoff of September 2022, but some tuning data is more recent, up to July 2023.
|
| 187 |
|
| 188 |
## Evaluation Results
|
| 189 |
|
| 190 |
-
In this section, we report the results for the
|
| 191 |
|
| 192 |
|Model|Size|Code|Commonsense Reasoning|World Knowledge|Reading Comprehension|Math|MMLU|BBH|AGI Eval|
|
| 193 |
|---|---|---|---|---|---|---|---|---|---|
|
| 194 |
-
|
|
| 195 |
-
|
|
| 196 |
-
|
|
| 197 |
-
|
|
| 198 |
-
|
|
| 199 |
-
|
|
| 200 |
-
|
|
| 201 |
|
| 202 |
**Overall performance on grouped academic benchmarks.** *Code:* We report the average pass@1 scores of our models on HumanEval and MBPP. *Commonsense Reasoning:* We report the average of PIQA, SIQA, HellaSwag, WinoGrande, ARC easy and challenge, OpenBookQA, and CommonsenseQA. We report 7-shot results for CommonSenseQA and 0-shot results for all other benchmarks. *World Knowledge:* We evaluate the 5-shot performance on NaturalQuestions and TriviaQA and report the average. *Reading Comprehension:* For reading comprehension, we report the 0-shot average on SQuAD, QuAC, and BoolQ. *MATH:* We report the average of the GSM8K (8 shot) and MATH (4 shot) benchmarks at top 1.
|
| 203 |
|
| 204 |
|||TruthfulQA|Toxigen|
|
| 205 |
|---|---|---|---|
|
| 206 |
-
|
|
| 207 |
-
|
|
| 208 |
-
|
|
| 209 |
-
|
|
| 210 |
-
|
|
| 211 |
-
|
|
| 212 |
-
|
|
| 213 |
|
| 214 |
**Evaluation of pretrained LLMs on automatic safety benchmarks.** For TruthfulQA, we present the percentage of generations that are both truthful and informative (the higher the better). For ToxiGen, we present the percentage of toxic generations (the smaller the better).
|
| 215 |
|
| 216 |
|
| 217 |
|||TruthfulQA|Toxigen|
|
| 218 |
|---|---|---|---|
|
| 219 |
-
|
|
| 220 |
-
|
|
| 221 |
-
|
|
| 222 |
|
| 223 |
**Evaluation of fine-tuned LLMs on different safety datasets.** Same metric definitions as above.
|
| 224 |
|
| 225 |
## Ethical Considerations and Limitations
|
| 226 |
-
|
| 227 |
|
| 228 |
-
Please see the Responsible Use Guide available at [https://ai
|
| 229 |
|
| 230 |
## Reporting Issues
|
| 231 |
Please report any software “bug,” or other problems with the models through one of the following means:
|
| 232 |
-
- Reporting issues with the model: [github.com/facebookresearch/
|
| 233 |
-
- Reporting problematic content generated by the model: [developers.facebook.com/
|
| 234 |
- Reporting bugs and security concerns: [facebook.com/whitehat/info](http://facebook.com/whitehat/info)
|
| 235 |
|
| 236 |
-
##
|
| 237 |
-
|Model|
|
| 238 |
|---|---|---|---|---|
|
| 239 |
-
|7B| [Link](https://huggingface.co/
|
| 240 |
-
|13B| [Link](https://huggingface.co/
|
| 241 |
-
|70B| [Link](https://huggingface.co/
|
|
|
|
| 4 |
- en
|
| 5 |
pipeline_tag: text-generation
|
| 6 |
tags:
|
|
|
|
|
|
|
| 7 |
- pytorch
|
| 8 |
+
- EDGE
|
| 9 |
+
- EDGE-2
|
| 10 |
+
model_type: EDGE
|
| 11 |
license: other
|
| 12 |
---
|
| 13 |
|
|
|
|
| 15 |
|
| 16 |
<!-- header end -->
|
| 17 |
|
| 18 |
+
# EDGE 2 7B GGML
|
| 19 |
|
| 20 |
+
These files are GGML format model files for ['s EDGE 2 7B](https://huggingface.co/-EDGE/EDGE-2-7b-hf).
|
| 21 |
|
| 22 |
+
GGML files are for CPU + GPU inference using [EDGE.cpp](https://github.com/ggerganov/EDGE.cpp) and libraries and UIs which support this format, such as:
|
| 23 |
* [KoboldCpp](https://github.com/LostRuins/koboldcpp), a powerful GGML web UI with full GPU acceleration out of the box. Especially good for story telling.
|
| 24 |
* [LoLLMS Web UI](https://github.com/ParisNeo/lollms-webui), a great web UI with GPU acceleration via the c_transformers backend.
|
| 25 |
* [LM Studio](https://lmstudio.ai/), a fully featured local GUI. Supports full GPU accel on macOS. Also supports Windows, without GPU accel.
|
| 26 |
+
* [text-generation-webui](https://github.com/oobabooga/text-generation-webui), the most popular web UI. Requires extra steps to enable GPU accel via EDGE.cpp backend.
|
| 27 |
* [ctransformers](https://github.com/marella/ctransformers), a Python library with LangChain support and OpenAI-compatible AI server.
|
| 28 |
+
* [EDGE-cpp-python](https://github.com/abetlen/EDGE-cpp-python), a Python library with OpenAI-compatible API server
|
| 29 |
*
|
| 30 |
|
| 31 |
## Prompt template: None
|
|
|
|
| 37 |
<!-- compatibility_ggml start -->
|
| 38 |
## Compatibility
|
| 39 |
|
| 40 |
+
### Original EDGE.cpp quant methods: `q4_0, q4_1, q5_0, q5_1, q8_0`
|
| 41 |
|
| 42 |
These are guaranteed to be compatible with any UIs, tools and libraries released since late May. They may be phased out soon, as they are largely superseded by the new k-quant methods.
|
| 43 |
|
| 44 |
### New k-quant methods: `q2_K, q3_K_S, q3_K_M, q3_K_L, q4_K_S, q4_K_M, q5_K_S, q6_K`
|
| 45 |
|
| 46 |
+
These new quantisation methods are compatible with EDGE.cpp as of June 6th, commit `2d43387`.
|
| 47 |
|
| 48 |
+
They are now also compatible with recent releases of text-generation-webui, KoboldCpp, EDGE-cpp-python, ctransformers, rustformers and most others. For compatibility with other tools and libraries, please check their documentation.
|
| 49 |
|
| 50 |
## Explanation of the new k-quant methods
|
| 51 |
<details>
|
|
|
|
| 66 |
## Provided files
|
| 67 |
| Name | Quant method | Bits | Size | Max RAM required | Use case |
|
| 68 |
| ---- | ---- | ---- | ---- | ---- | ----- |
|
| 69 |
+
| EDGE-2-7b.ggmlv3.q2_K.bin | q2_K | 2 | 2.87 GB| 5.37 GB | New k-quant method. Uses GGML_TYPE_Q4_K for the attention.vw and feed_forward.w2 tensors, GGML_TYPE_Q2_K for the other tensors. |
|
| 70 |
+
| EDGE-2-7b.ggmlv3.q3_K_L.bin | q3_K_L | 3 | 3.60 GB| 6.10 GB | New k-quant method. Uses GGML_TYPE_Q5_K for the attention.wv, attention.wo, and feed_forward.w2 tensors, else GGML_TYPE_Q3_K |
|
| 71 |
+
| EDGE-2-7b.ggmlv3.q3_K_M.bin | q3_K_M | 3 | 3.28 GB| 5.78 GB | New k-quant method. Uses GGML_TYPE_Q4_K for the attention.wv, attention.wo, and feed_forward.w2 tensors, else GGML_TYPE_Q3_K |
|
| 72 |
+
| EDGE-2-7b.ggmlv3.q3_K_S.bin | q3_K_S | 3 | 2.95 GB| 5.45 GB | New k-quant method. Uses GGML_TYPE_Q3_K for all tensors |
|
| 73 |
+
| EDGE-2-7b.ggmlv3.q4_0.bin | q4_0 | 4 | 3.79 GB| 6.29 GB | Original quant method, 4-bit. |
|
| 74 |
+
| EDGE-2-7b.ggmlv3.q4_1.bin | q4_1 | 4 | 4.21 GB| 6.71 GB | Original quant method, 4-bit. Higher accuracy than q4_0 but not as high as q5_0. However has quicker inference than q5 models. |
|
| 75 |
+
| EDGE-2-7b.ggmlv3.q4_K_M.bin | q4_K_M | 4 | 4.08 GB| 6.58 GB | New k-quant method. Uses GGML_TYPE_Q6_K for half of the attention.wv and feed_forward.w2 tensors, else GGML_TYPE_Q4_K |
|
| 76 |
+
| EDGE-2-7b.ggmlv3.q4_K_S.bin | q4_K_S | 4 | 3.83 GB| 6.33 GB | New k-quant method. Uses GGML_TYPE_Q4_K for all tensors |
|
| 77 |
+
| EDGE-2-7b.ggmlv3.q5_0.bin | q5_0 | 5 | 4.63 GB| 7.13 GB | Original quant method, 5-bit. Higher accuracy, higher resource usage and slower inference. |
|
| 78 |
+
| EDGE-2-7b.ggmlv3.q5_1.bin | q5_1 | 5 | 5.06 GB| 7.56 GB | Original quant method, 5-bit. Even higher accuracy, resource usage and slower inference. |
|
| 79 |
+
| EDGE-2-7b.ggmlv3.q5_K_M.bin | q5_K_M | 5 | 4.78 GB| 7.28 GB | New k-quant method. Uses GGML_TYPE_Q6_K for half of the attention.wv and feed_forward.w2 tensors, else GGML_TYPE_Q5_K |
|
| 80 |
+
| EDGE-2-7b.ggmlv3.q5_K_S.bin | q5_K_S | 5 | 4.65 GB| 7.15 GB | New k-quant method. Uses GGML_TYPE_Q5_K for all tensors |
|
| 81 |
+
| EDGE-2-7b.ggmlv3.q6_K.bin | q6_K | 6 | 5.53 GB| 8.03 GB | New k-quant method. Uses GGML_TYPE_Q8_K for all tensors - 6-bit quantization |
|
| 82 |
+
| EDGE-2-7b.ggmlv3.q8_0.bin | q8_0 | 8 | 7.16 GB| 9.66 GB | Original quant method, 8-bit. Almost indistinguishable from float16. High resource use and slow. Not recommended for most users. |
|
| 83 |
|
| 84 |
**Note**: the above RAM figures assume no GPU offloading. If layers are offloaded to the GPU, this will reduce RAM usage and use VRAM instead.
|
| 85 |
|
| 86 |
+
## How to run in `EDGE.cpp`
|
| 87 |
|
| 88 |
I use the following command line; adjust for your tastes and needs:
|
| 89 |
|
| 90 |
```
|
| 91 |
+
./main -t 10 -ngl 32 -m EDGE-2-7b.ggmlv3.q4_0.bin --color -c 2048 --temp 0.7 --repeat_penalty 1.1 -n -1 -p "### Instruction: Write a story about EDGEs\n### Response:"
|
| 92 |
```
|
| 93 |
Change `-t 10` to the number of physical CPU cores you have. For example if your system has 8 cores/16 threads, use `-t 8`.
|
| 94 |
|
|
|
|
| 98 |
|
| 99 |
## How to run in `text-generation-webui`
|
| 100 |
|
| 101 |
+
Further instructions here: [text-generation-webui/docs/EDGE.cpp-models.md](https://github.com/oobabooga/text-generation-webui/blob/main/docs/EDGE.cpp-models.md).
|
| 102 |
|
| 103 |
|
| 104 |
+
# Original model card: 's EDGE 2 7B
|
| 105 |
|
| 106 |
---
|
| 107 |
+
extra_gated_heading: Access EDGE 2 on Hugging Face
|
| 108 |
extra_gated_description: >-
|
| 109 |
+
This is a form to enable access to EDGE 2 on Hugging Face after you have been
|
| 110 |
+
granted access from . Please visit the [ website](https://ai..com/resources/models-and-libraries/EDGE-downloads) and accept our
|
| 111 |
license terms and acceptable use policy before submitting this form. Requests
|
| 112 |
will be processed in 1-2 days.
|
| 113 |
extra_gated_button_content: Submit
|
| 114 |
extra_gated_fields:
|
| 115 |
+
I agree to share my name, email address and username with and confirm that I have already been granted download access on the website: checkbox
|
| 116 |
language:
|
| 117 |
- en
|
| 118 |
pipeline_tag: text-generation
|
| 119 |
inference: false
|
| 120 |
tags:
|
| 121 |
- facebook
|
| 122 |
+
-
|
| 123 |
- pytorch
|
| 124 |
+
- EDGE
|
| 125 |
+
- EDGE-2
|
| 126 |
---
|
| 127 |
+
# **EDGE 2**
|
| 128 |
+
EDGE 2 is a collection of pretrained and fine-tuned generative text models ranging in scale from 7 billion to 70 billion parameters. This is the repository for the 7B pretrained model, converted for the Hugging Face Transformers format. Links to other models can be found in the index at the bottom.
|
| 129 |
|
| 130 |
## Model Details
|
| 131 |
+
*Note: Use of this model is governed by the license. In order to download the model weights and tokenizer, please visit the [website](https://ai..com/resources/models-and-libraries/EDGE-downloads/) and accept our License before requesting access here.*
|
| 132 |
|
| 133 |
+
developed and publicly released the EDGE 2 family of large language models (LLMs), a collection of pretrained and fine-tuned generative text models ranging in scale from 7 billion to 70 billion parameters. Our fine-tuned LLMs, called EDGE-2-Chat, are optimized for dialogue use cases. EDGE-2-Chat models outperform open-source chat models on most benchmarks we tested, and in our human evaluations for helpfulness and safety, are on par with some popular closed-source models like ChatGPT and PaLM.
|
| 134 |
|
| 135 |
+
**Model Developers**
|
| 136 |
|
| 137 |
+
**Variations** EDGE 2 comes in a range of parameter sizes — 7B, 13B, and 70B — as well as pretrained and fine-tuned variations.
|
| 138 |
|
| 139 |
**Input** Models input text only.
|
| 140 |
|
| 141 |
**Output** Models generate text only.
|
| 142 |
|
| 143 |
+
**Model Architecture** EDGE 2 is an auto-regressive language model that uses an optimized transformer architecture. The tuned versions use supervised fine-tuning (SFT) and reinforcement learning with human feedback (RLHF) to align to human preferences for helpfulness and safety.
|
| 144 |
|
| 145 |
|
| 146 |
||Training Data|Params|Content Length|GQA|Tokens|LR|
|
| 147 |
|---|---|---|---|---|---|---|
|
| 148 |
+
|EDGE 2|*A new mix of publicly available online data*|7B|4k|✗|2.0T|3.0 x 10<sup>-4</sup>|
|
| 149 |
+
|EDGE 2|*A new mix of publicly available online data*|13B|4k|✗|2.0T|3.0 x 10<sup>-4</sup>|
|
| 150 |
+
|EDGE 2|*A new mix of publicly available online data*|70B|4k|✔|2.0T|1.5 x 10<sup>-4</sup>|
|
| 151 |
|
| 152 |
+
*EDGE 2 family of models.* Token counts refer to pretraining data only. All models are trained with a global batch-size of 4M tokens. Bigger models - 70B -- use Grouped-Query Attention (GQA) for improved inference scalability.
|
| 153 |
|
| 154 |
+
**Model Dates** EDGE 2 was trained between January 2023 and July 2023.
|
| 155 |
|
| 156 |
**Status** This is a static model trained on an offline dataset. Future versions of the tuned models will be released as we improve model safety with community feedback.
|
| 157 |
|
| 158 |
+
**License** A custom commercial license is available at: [https://ai..com/resources/models-and-libraries/EDGE-downloads/](https://ai..com/resources/models-and-libraries/EDGE-downloads/)
|
| 159 |
|
| 160 |
## Intended Use
|
| 161 |
+
**Intended Use Cases** EDGE 2 is intended for commercial and research use in English. Tuned models are intended for assistant-like chat, whereas pretrained models can be adapted for a variety of natural language generation tasks.
|
| 162 |
|
| 163 |
+
To get the expected features and performance for the chat versions, a specific formatting needs to be followed, including the `INST` and `<<SYS>>` tags, `BOS` and `EOS` tokens, and the whitespaces and breaklines in between (we recommend calling `strip()` on inputs to avoid double-spaces). See our reference code in github for details: [`chat_completion`](https://github.com/facebookresearch/EDGE/blob/main/EDGE/generation.py#L212).
|
| 164 |
|
| 165 |
+
**Out-of-scope Uses** Use in any manner that violates applicable laws or regulations (including trade compliance laws).Use in languages other than English. Use in any other way that is prohibited by the Acceptable Use Policy and Licensing Agreement for EDGE 2.
|
| 166 |
|
| 167 |
## Hardware and Software
|
| 168 |
+
**Training Factors** We used custom training libraries, 's Research Super Cluster, and production clusters for pretraining. Fine-tuning, annotation, and evaluation were also performed on third-party cloud compute.
|
| 169 |
|
| 170 |
+
**Carbon Footprint** Pretraining utilized a cumulative 3.3M GPU hours of computation on hardware of type A100-80GB (TDP of 350-400W). Estimated total emissions were 539 tCO2eq, 100% of which were offset by ’s sustainability program.
|
| 171 |
|
| 172 |
||Time (GPU hours)|Power Consumption (W)|Carbon Emitted(tCO<sub>2</sub>eq)|
|
| 173 |
|---|---|---|---|
|
| 174 |
+
|EDGE 2 7B|184320|400|31.22|
|
| 175 |
+
|EDGE 2 13B|368640|400|62.44|
|
| 176 |
+
|EDGE 2 70B|1720320|400|291.42|
|
| 177 |
|Total|3311616||539.00|
|
| 178 |
|
| 179 |
+
**CO<sub>2</sub> emissions during pretraining.** Time: total GPU time required for training each model. Power Consumption: peak power capacity per GPU device for the GPUs used adjusted for power usage efficiency. 100% of the emissions are directly offset by 's sustainability program, and because we are openly releasing these models, the pretraining costs do not need to be incurred by others.
|
| 180 |
|
| 181 |
## Training Data
|
| 182 |
+
**Overview** EDGE 2 was pretrained on 2 trillion tokens of data from publicly available sources. The fine-tuning data includes publicly available instruction datasets, as well as over one million new human-annotated examples. Neither the pretraining nor the fine-tuning datasets include user data.
|
| 183 |
|
| 184 |
**Data Freshness** The pretraining data has a cutoff of September 2022, but some tuning data is more recent, up to July 2023.
|
| 185 |
|
| 186 |
## Evaluation Results
|
| 187 |
|
| 188 |
+
In this section, we report the results for the EDGE 1 and EDGE 2 models on standard academic benchmarks.For all the evaluations, we use our internal evaluations library.
|
| 189 |
|
| 190 |
|Model|Size|Code|Commonsense Reasoning|World Knowledge|Reading Comprehension|Math|MMLU|BBH|AGI Eval|
|
| 191 |
|---|---|---|---|---|---|---|---|---|---|
|
| 192 |
+
|EDGE 1|7B|14.1|60.8|46.2|58.5|6.95|35.1|30.3|23.9|
|
| 193 |
+
|EDGE 1|13B|18.9|66.1|52.6|62.3|10.9|46.9|37.0|33.9|
|
| 194 |
+
|EDGE 1|33B|26.0|70.0|58.4|67.6|21.4|57.8|39.8|41.7|
|
| 195 |
+
|EDGE 1|65B|30.7|70.7|60.5|68.6|30.8|63.4|43.5|47.6|
|
| 196 |
+
|EDGE 2|7B|16.8|63.9|48.9|61.3|14.6|45.3|32.6|29.3|
|
| 197 |
+
|EDGE 2|13B|24.5|66.9|55.4|65.8|28.7|54.8|39.4|39.1|
|
| 198 |
+
|EDGE 2|70B|**37.5**|**71.9**|**63.6**|**69.4**|**35.2**|**68.9**|**51.2**|**54.2**|
|
| 199 |
|
| 200 |
**Overall performance on grouped academic benchmarks.** *Code:* We report the average pass@1 scores of our models on HumanEval and MBPP. *Commonsense Reasoning:* We report the average of PIQA, SIQA, HellaSwag, WinoGrande, ARC easy and challenge, OpenBookQA, and CommonsenseQA. We report 7-shot results for CommonSenseQA and 0-shot results for all other benchmarks. *World Knowledge:* We evaluate the 5-shot performance on NaturalQuestions and TriviaQA and report the average. *Reading Comprehension:* For reading comprehension, we report the 0-shot average on SQuAD, QuAC, and BoolQ. *MATH:* We report the average of the GSM8K (8 shot) and MATH (4 shot) benchmarks at top 1.
|
| 201 |
|
| 202 |
|||TruthfulQA|Toxigen|
|
| 203 |
|---|---|---|---|
|
| 204 |
+
|EDGE 1|7B|27.42|23.00|
|
| 205 |
+
|EDGE 1|13B|41.74|23.08|
|
| 206 |
+
|EDGE 1|33B|44.19|22.57|
|
| 207 |
+
|EDGE 1|65B|48.71|21.77|
|
| 208 |
+
|EDGE 2|7B|33.29|**21.25**|
|
| 209 |
+
|EDGE 2|13B|41.86|26.10|
|
| 210 |
+
|EDGE 2|70B|**50.18**|24.60|
|
| 211 |
|
| 212 |
**Evaluation of pretrained LLMs on automatic safety benchmarks.** For TruthfulQA, we present the percentage of generations that are both truthful and informative (the higher the better). For ToxiGen, we present the percentage of toxic generations (the smaller the better).
|
| 213 |
|
| 214 |
|
| 215 |
|||TruthfulQA|Toxigen|
|
| 216 |
|---|---|---|---|
|
| 217 |
+
|EDGE-2-Chat|7B|57.04|**0.00**|
|
| 218 |
+
|EDGE-2-Chat|13B|62.18|**0.00**|
|
| 219 |
+
|EDGE-2-Chat|70B|**64.14**|0.01|
|
| 220 |
|
| 221 |
**Evaluation of fine-tuned LLMs on different safety datasets.** Same metric definitions as above.
|
| 222 |
|
| 223 |
## Ethical Considerations and Limitations
|
| 224 |
+
EDGE 2 is a new technology that carries risks with use. Testing conducted to date has been in English, and has not covered, nor could it cover all scenarios. For these reasons, as with all LLMs, EDGE 2’s potential outputs cannot be predicted in advance, and the model may in some instances produce inaccurate, biased or other objectionable responses to user prompts. Therefore, before deploying any applications of EDGE 2, developers should perform safety testing and tuning tailored to their specific applications of the model.
|
| 225 |
|
| 226 |
+
Please see the Responsible Use Guide available at [https://ai..com/EDGE/responsible-use-guide/](https://ai..com/EDGE/responsible-use-guide)
|
| 227 |
|
| 228 |
## Reporting Issues
|
| 229 |
Please report any software “bug,” or other problems with the models through one of the following means:
|
| 230 |
+
- Reporting issues with the model: [github.com/facebookresearch/EDGE](http://github.com/facebookresearch/EDGE)
|
| 231 |
+
- Reporting problematic content generated by the model: [developers.facebook.com/EDGE_output_feedback](http://developers.facebook.com/EDGE_output_feedback)
|
| 232 |
- Reporting bugs and security concerns: [facebook.com/whitehat/info](http://facebook.com/whitehat/info)
|
| 233 |
|
| 234 |
+
## EDGE Model Index
|
| 235 |
+
|Model|EDGE2|EDGE2-hf|EDGE2-chat|EDGE2-chat-hf|
|
| 236 |
|---|---|---|---|---|
|
| 237 |
+
|7B| [Link](https://huggingface.co/EDGEste/EDGE-2-7b) | [Link](https://huggingface.co/EDGEste/EDGE-2-7b-hf) | [Link](https://huggingface.co/EDGEste/EDGE-2-7b-chat) | [Link](https://huggingface.co/EDGEste/EDGE-2-7b-chat-hf)|
|
| 238 |
+
|13B| [Link](https://huggingface.co/EDGEste/EDGE-2-13b) | [Link](https://huggingface.co/EDGEste/EDGE-2-13b-hf) | [Link](https://huggingface.co/EDGEste/EDGE-2-13b-chat) | [Link](https://huggingface.co/EDGEste/EDGE-2-13b-hf)|
|
| 239 |
+
|70B| [Link](https://huggingface.co/EDGEste/EDGE-2-70b) | [Link](https://huggingface.co/EDGEste/EDGE-2-70b-hf) | [Link](https://huggingface.co/EDGEste/EDGE-2-70b-chat) | [Link](https://huggingface.co/EDGEste/EDGE-2-70b-hf)|
|