LMFlow
LMFlow copied to clipboard
[usability] Accelerate Support
Description
- LMFlow now defaults to using Accelerate (i.e, run scripts using
accelerate launch ... finetune.py ...). If you prefer to use deepspeed (deepspeed ... finetune.py ...) or accelerate + deepspeed backend, please install usingpip install -e '.[deepspeed]' - Removed/archived some less frequently used docs/scripts/modules.
-
docker -
scripts/data_preprocess,scripts/speculative_decoding,scripts/tools,scripts/vocab_extension -
service -
utils
-
- LISA callback and
create_customized_optimizerare separated fromfinetuner.py - Add unittest for
finetuner.py
Known Issues
-
hf_encoder_decoder_modelis disabled temporarily. We will have a comprehensive update for encoder-decoder model later.
Tests
| scripts | Single GPU | Multiple GPUs |
|---|---|---|
| run_finetune.sh | ✅ | ✅ |
| run_finetune_with_lora.sh | ✅ | ✅ |
| run_finetune_with_qlora.sh | ✅ | ✅ |
| run_finetune_with_lisa.sh | ✅ | ⌛ |
| run_finetune_with_custom_optim.sh | ✅ | ✅ |