EasyLLM

Built upon Megatron-Deepspeed and HuggingFace Trainer, EasyLLM has reorganized the code logic with a focus on usability. While enhancing usability, it also ensures training efficiency.

Install

• Install python requirements

  pip install -r requirements.txt

  other dependency

  • flash-attn (dropout_layer_norm) (maybe you need to compile it by yourself)

• Pull deepspeed & add them to pythonpath

  export PYTHONPATH=/path/to/DeepSpeed:$PYTHONPATH

• Install package in development mode

  pip install -e . -v

Train

Train Example

Infer and Eval

Infer Example

Support Models

• qwen14b,
• internlm7-20b,
• baichuan1/2 (7b-13b)
• llama1-2 (7b/13b/70b)

Model Example

Data

Data Example

3D Parallel config setting

Parallel Example

Speed Benchmark

Speed Benchmark

Dynamic Checkpoint

To optimize the model training performance in terms of time and space, EasyLLM supports Dynamic Checkpoint. Based on the input token size, it enables checkpointing for some layers. The configuration file settings are as follows:

Dynamic Checkpoint Example

License

This repository is released under the Apache-2.0 license.

Acknowledgement

We learned a lot from the following projects when developing EasyLLM.

• DeepSpeed
• Megatron-DeepSpeed
• Megatron-LM
• Flash Attention 1&2
• LightLLM
• Huggingface Transformers