upgrade accelerate to 1.0.1 (#1969)

axolotl-ai-cloud · Oct 14, 2024 · 335027f · 335027f
1 parent ec4272c
commit 335027f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -4,7 +4,7 @@ peft==0.13.2
 transformers==4.45.2
 tokenizers>=0.20.1
 bitsandbytes==0.44.1
-accelerate==0.34.2
+accelerate==1.0.1
 datasets==3.0.1
 deepspeed==0.14.4
 pydantic==2.6.3

diff --git a/src/axolotl/train.py b/src/axolotl/train.py
@@ -10,7 +10,6 @@
 
 import torch
 import transformers.modelcard
-from accelerate import Accelerator
 from accelerate.logging import get_logger
 from accelerate.utils import save_fsdp_model
 from datasets import Dataset
@@ -97,8 +96,6 @@ def train(
     if cfg.adapter:
         msg += " and peft_config..."
     LOG.debug(msg)
-    # we wait unitl the last possible moment to setup Accelerator
-    Accelerator()
     model, peft_config = load_model(
         cfg, tokenizer, processor=processor, inference=cli_args.inference
     )