remove import

This commit is contained in:
S1ro1
2025-07-07 13:30:53 +00:00
parent 53ef524773
commit cd31b02bfa

View File

@ -25,7 +25,7 @@ from torch.utils.data import DataLoader
from transformers import AutoModelForSequenceClassification, AutoTokenizer, get_linear_schedule_with_warmup
from accelerate import Accelerator, DistributedType
from accelerate.utils import SAFE_WEIGHTS_NAME, ParallelismConfig, TorchTensorParallelKwargs, set_seed
from accelerate.utils import SAFE_WEIGHTS_NAME, ParallelismConfig, set_seed
from accelerate.utils.deepspeed import DummyOptim, DummyScheduler