Update README_en.md

This commit is contained in:
s-JoL 2023-04-11 15:53:05 +08:00 committed by GitHub
parent be2f0960c7
commit ce8bc5249f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -104,6 +104,7 @@ Check the DataLoader output with the following command:
```bash
python3 dataset/pretrain_dataset.py
```
Verification of data intergrity can be foud in this [issue]((https://github.com/s-JoL/Open-Llama/issues/5)
### Model Structure
We modified the [Llama](https://github.com/facebookresearch/llama) model in the Transformers library based on section 2.4 "Efficient Implementation" in the original paper and introduced some optimizations from other papers. Specifically, we introduced the memory_efficient_attention operation from the [xformers library](https://github.com/facebookresearch/xformers) by META for computing self-attention, which significantly improves performance by about 30%. Please refer to modeling_llama.py for details.
@ -261,4 +262,4 @@ Build an online demo using Gradio.
year={2023},
howpublished={\url{https://github.com/Bayes-Song/Open-Llama}},
}
```
```