upgrade accelerate to 1.0.1 (#1969)

This commit is contained in:
Wing Lian
2024-10-13 20:04:30 -04:00
committed by GitHub
parent ec4272c3a0
commit 335027f155
2 changed files with 1 additions and 4 deletions

View File

@@ -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

View File

@@ -10,7 +10,6 @@ from typing import Optional, Tuple, Union
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
)