LMFlow icon indicating copy to clipboard operation
LMFlow copied to clipboard

[usability] Accelerate Support

Open wheresmyhair opened this issue 10 months ago • 0 comments

Description

  1. 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 using pip install -e '.[deepspeed]'
  2. Removed/archived some less frequently used docs/scripts/modules.
    • docker
    • scripts/data_preprocess, scripts/speculative_decoding, scripts/tools, scripts/vocab_extension
    • service
    • utils
  3. LISA callback and create_customized_optimizer are separated from finetuner.py
  4. Add unittest for finetuner.py

Known Issues

  1. hf_encoder_decoder_model is 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

wheresmyhair avatar Mar 04 '25 09:03 wheresmyhair